Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dimage_ops.cc131 TensorShape channel_shape = input_shape; in Compile() local
264 TensorShape channel_shape = input_shape; in Compile() local
318 TensorShape channel_shape = input_shape; in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper_test.cc141 TensorShape channel_shape({num_channels}); in TEST_F() local
231 TensorShape channel_shape({num_channels}); in TEST_F() local
373 TensorShape channel_shape({num_channels}); in TEST_F() local
469 TensorShape channel_shape({num_channels}); in TEST_F() local
H A Dmkl_remapper_test.cc321 TensorShape channel_shape({num_channels}); in TEST_F() local