Method InputShape

tensorflow/compiler/tf2xla/xla_op_kernel.cc:88–90

}

TensorShape XlaOpKernelContext::InputShape(int index) {
  return context_->input(index).shape();
}

TensorShape XlaOpKernelContext::InputShape(absl::string_view name) {
  return GetInputTensorByName(name).shape();