| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| batchtospace_op.cc | 30 input_tensor_shape.dim_sizes();
|
| dynamic_stitch_op.cc | 166 input[input_num] = xla::Reshape(handle, new_shape.dim_sizes());
|
| gather_op.cc | 79 xla::Broadcast(XlaHelpers::Zero(builder, dtype), out_shape.dim_sizes());
|
| spacetobatch_op.cc | 30 input_tensor_shape.dim_sizes();
|
| pooling_ops.cc | 232 input_shape.dim_sizes(), ksize, stride, padding_, xla_data_format);
432 MakeSpatialPadding(gradients_shape.dim_sizes(), ksize_, stride_int64s,
435 xla::AvgPoolGrad(converted_out_backprop, gradients_shape.dim_sizes(),
|
| image_ops.cc | 49 auto zeros = xla::Broadcast(zero, shape.dim_sizes());
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| tensor_slice_dataset_op.cc | 140 TensorShape(dataset()->shapes_[i].dim_sizes()));
|
| /external/tensorflow/tensorflow/core/kernels/ |
| debug_ops.h | 377 static_cast<double>(input_shape.dim_sizes()[d]);
|
| list_kernels.h | 313 gtl::ArraySlice<int64>(element_shape.dim_sizes()).subspan(1));
376 gtl::ArraySlice<int64>(t.shape().dim_sizes()).subspan(1));
523 PartialTensorShape(element_shape.dim_sizes());
[all...] |
| padding_fifo_queue.cc | 404 for (int64 s : partial.dim_sizes()) shape.AddDim(s < 0 ? 0 : s);
|
| quantized_add_op_test.cc | 130 for (const int64 dim : expected_values_tensor.shape().dim_sizes()) {
|
| quantized_mul_op_test.cc | 130 for (const int64 dim : expected_values_tensor.shape().dim_sizes()) {
|
| sparse_reduce_op.cc | 322 auto out_dim_sizes = reduction.reduced_shape.dim_sizes();
|
| tensor_array.h | 481 element_shape_ = PartialTensorShape(value_t->shape().dim_sizes());
|
| conv_ops_fused_image_transform.cc | 421 const int64 input_height = input.shape().dim_sizes()[1];
422 task_params.input_width = input.shape().dim_sizes()[2];
[all...] |
| functional_ops.cc | 643 for (int64 d : partial_shape.dim_sizes()) {
|
| example_parsing_ops.cc | 624 for (const int dim : attrs_.context_dense_shapes[d].dim_sizes())
[all...] |
| remote_fused_graph_execute_utils.cc | 633 for (const int64 dim : shape.dim_sizes()) {
659 for (const int64 dim : shape.dim_sizes()) {
[all...] |
| /external/tensorflow/tensorflow/contrib/periodic_resample/kernels/ |
| periodic_resample_op.h | 286 process_desired_shape(context, original_shape, desired_shape.dim_sizes(),
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_op_kernel.cc | 116 index, context_->input(index).shape().dim_sizes(), constant_literal);
399 *value = xla::Reshape(variable->value(), variable->shape().dim_sizes());
|
| /external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
| fused_conv2d_bias_activation_op_test_base.py | 167 def _IotaNdF32Constant(dim_sizes):
174 return constant_op.constant(MakeList(dim_sizes), dtype=dtypes.float32)
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| shape_inference_test.cc | [all...] |
| tensor.cc | [all...] |
| /external/tensorflow/tensorflow/core/util/ |
| example_proto_helper.cc | 349 for (const int dim : shape.dim_sizes()) out_shape.AddDim(dim);
|
| example_proto_fast_parsing.cc | [all...] |
