| /external/tensorflow/tensorflow/core/kernels/ |
| cudnn_pooling_gpu.cc | 43 const int64 in_batch = GetTensorDim(tensor_in, data_format, 'N'); local
73 input_desc.set_count(in_batch)
77 output_desc.set_count(in_batch)
130 const int64 in_batch = GetTensorDim(tensor_in_shape, data_format, 'N'); local
193 orig_output_desc.set_count(in_batch)
198 orig_input_desc.set_count(in_batch)
|
| pooling_ops_3d.cc | 172 const int64 in_batch = GetTensorDim(tensor_in, data_format_, 'N'); variable
190 TensorShape out_shape = ShapeFromFormat(data_format_, in_batch,
[all...] |
| conv_ops_3d.cc | 104 const int64 in_batch = GetTensorDim(input, data_format_, 'N'); variable
126 data_format_, in_batch, {{out[0], out[1], out[2]}}, out_depth);
175 const int64 in_batch = GetTensorDim(input, data_format, 'N'); local
205 const uint64 m = in_batch * in_planes * in_rows * in_cols;
232 const uint64 m = in_batch;
272 data_format, in_batch, {{new_in_planes, new_in_rows, new_in_cols}},
291 FORMAT_NCHW, in_batch, {{in_planes, in_rows, in_cols}}, in_depth);
313 input_desc.set_count(in_batch)
320 output_desc.set_count(in_batch)
356 ShapeFromFormat(FORMAT_NCHW, in_batch,
[all...] |
| fractional_avg_pool_op.cc | 253 const int64 in_batch = orig_input_tensor_shape_flat(0); variable
274 in_cols * in_rows * in_batch);
|
| conv_ops.cc | 384 // [ in_batch, out_rows, out_cols, out_depth ]
560 const int64 in_batch = GetTensorDim(input, data_format, 'N'); local
594 ShapeFromFormat(data_format, in_batch, new_in_rows,
612 ShapeFromFormat(FORMAT_NCHW, in_batch, in_rows, in_cols, in_depths);
632 input_desc.set_count(in_batch)
691 in_batch, // batch
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| algebraic_simplifier_test.cc | 1773 int in_batch = 10; member in struct:ConvTestOptions
[all...] |
