Home
last modified time | relevance | path

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

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