Searched defs:InputXlaShape (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ | ||
| H A D | xla_op_kernel.cc | 87 StatusOr<xla::Shape> XlaOpKernelContext::InputXlaShape(int index) { in InputXlaShape() function in tensorflow::XlaOpKernelContext |