Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dimage_ops.cc124 TensorShape channel_shape = input_shape; in Compile() local
257 TensorShape channel_shape = input_shape; in Compile() local
311 TensorShape channel_shape = input_shape; in Compile() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmkl_remapper_test.cc322 TensorShape channel_shape({num_channels}); in TEST_F() local
Dremapper_test.cc138 TensorShape channel_shape({num_channels}); in TEST_F() local
245 TensorShape channel_shape({num_channels}); in TEST_F() local