Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dlight_outside_compilation_kernels_for_test.cc134 TensorShape allocated_shape; in Compute() local
325 TensorShape allocated_shape; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_slice_op.cc68 TensorShape allocated_shape; in operator ()() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dlight_outside_compilation.cc376 TensorShape allocated_shape = allocated->shape(); in PopulateMetadataBufferIfNeeded() local