HomeSort by relevance Sort by last modified time
    Searched refs:AddDim (Results 76 - 100 of 112) sorted by null

1 2 34 5

  /external/tensorflow/tensorflow/core/kernels/
example_parsing_ops.cc 402 lengths_shape.AddDim(serialized_t.size());
625 out_shape.AddDim(dim);
    [all...]
quantized_concat_op.cc 209 output_shape.AddDim(output_concat_dim);
transpose_op.cc 173 shape.AddDim(dim_size);
batch_matmul_op_impl.h 560 out_shape.AddDim(in0.dim_size(i));
578 out_shape.AddDim(d0);
579 out_shape.AddDim(d3);
resource_variable_ops.cc 646 result_shape.AddDim(params.dim_size(i));
649 result_shape.AddDim(indices.dim_size(i));
652 result_shape.AddDim(params.dim_size(i));
    [all...]
set_kernels.cc 473 output_shape.AddDim(max_set_size);
555 output_shape.AddDim(max_set_size);
678 output_shape.AddDim(max_set_size);
mkl_conv_ops.cc 635 mkl_output_tf_shape.AddDim(
657 mkl_filter_output_tf_shape.AddDim(
    [all...]
fractional_avg_pool_op.cc 262 in_shape.AddDim(orig_input_tensor_shape_flat(i));
slice_op.cc 115 output_shape->AddDim(s);
decode_proto_op.cc 735 sizes_shape.AddDim(field_count);
792 out_shape.AddDim(max_sizes[fi]);
    [all...]
mkl_pooling_ops_common.h 657 output_tf_shape.AddDim(this->GetNumTElements(dst_pd));
708 output_tf_shape.AddDim(this->GetNumTElements(dst_pd));
functional_ops.cc 644 shape.AddDim(d == -1 ? 0 : d);
mkl_slice_op.cc 465 output_tf_shape.AddDim(output_pd->get_size() / sizeof(T));
pad_op.cc 95 output_shape.AddDim(before_d + size_d + after_d);
  /external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/
mpi_ops.cc     [all...]
  /external/tensorflow/tensorflow/contrib/mpi_collectives/
mpi_ops.cc     [all...]
  /external/tensorflow/tensorflow/core/util/
mkl_util.h 810 output_shape.AddDim(mkl_shape.GetSizes()[mkl_shape.tf_dim_idx(j)]);
    [all...]
  /external/tensorflow/tensorflow/contrib/periodic_resample/kernels/
periodic_resample_op.h 299 output_shape.AddDim(target_dimensions[i]);
  /external/tensorflow/tensorflow/core/framework/
tensor_shape.h 189 void AddDim(int64 size);
267 // Used by AddDim and MakeShapeHelper. Does no error checking.
289 /// zero dimensions and one element, and call AddDim() to add dimensions later.
  /external/tensorflow/tensorflow/core/kernels/data/experimental/
parse_example_dataset_op.cc 62 dense_shape.AddDim(dense_shapes_[i].dim_size(d));
  /external/tensorflow/tensorflow/core/util/sparse/
sparse_tensor_test.cc 716 shape.AddDim(1000);
759 shape.AddDim(1000);
  /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
stats_ops.cc 342 output_shape.AddDim(ready_to_split.size());
  /external/tensorflow/tensorflow/core/grappler/
grappler_item_builder.cc 490 shape.AddDim(size);
  /external/tensorflow/tensorflow/core/kernels/hexagon/
hexagon_control_wrapper.cc 363 output_shape.AddDim(dim.size());
  /external/tensorflow/tensorflow/python/lib/core/
py_func.cc 370 shape.AddDim(PyArray_SHAPE(input)[i]);

Completed in 2921 milliseconds

1 2 34 5