Home
last modified time | relevance | path

Searched refs: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
125 channel_shape.set_dim(channel_dim, 1); in Compile()
127 channel_shape); in Compile()
257 TensorShape channel_shape = input_shape; in Compile() local
258 channel_shape.set_dim(channel_dim, 1); in Compile()
260 RGBToHSV(context, b, {red, green, blue}, DT_FLOAT, channel_shape); in Compile()
311 TensorShape channel_shape = input_shape; in Compile() local
312 channel_shape.set_dim(channel_dim, 1); in Compile()
314 RGBToHSV(context, b, {red, green, blue}, DT_FLOAT, channel_shape); in Compile()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper_test.cc138 TensorShape channel_shape({num_channels}); in TEST_F() local
159 auto scale_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
160 auto offset_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
162 : channel_shape); in TEST_F()
164 : channel_shape); in TEST_F()
245 TensorShape channel_shape({num_channels}); in TEST_F() local
271 auto scale_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
272 auto offset_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
274 : channel_shape); in TEST_F()
276 : channel_shape); in TEST_F()
Dmkl_remapper_test.cc322 TensorShape channel_shape({num_channels}); in TEST_F() local
355 auto scale_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
356 auto offset_t = GenerateRandomTensor<DT_FLOAT>(channel_shape); in TEST_F()
358 : channel_shape); in TEST_F()
360 : channel_shape); in TEST_F()