Home
last modified time | relevance | path

Searched defs:InputShape (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.cc87 TensorShape XlaOpKernelContext::InputShape(int index) { in InputShape() function in tensorflow::XlaOpKernelContext
91 TensorShape XlaOpKernelContext::InputShape(absl::string_view name) { in InputShape() function in tensorflow::XlaOpKernelContext