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

12 3 4 5

  /external/tensorflow/tensorflow/core/framework/
tensor_shape.cc 130 AddDim(d.size());
225 AddDim(internal::SubtleMustCopy(s));
323 void TensorShapeBase<Shape>::AddDim(int64 size) {
384 for (auto d : shape) AddDim(d.size);
398 AddDim(dval);
431 AddDim(dval);
452 AddDim(dval);
636 out.AddDim(size);
646 for (auto dim : shape) out.AddDim(dim.size);
676 result->AddDim(dim0 >= 0 ? dim0 : dim1)
    [all...]
tensor_slice.cc 249 result_shape->AddDim(shape.dim_size(d));
256 result_shape->AddDim(length(d));
  /external/tensorflow/tensorflow/core/kernels/
mkl_batch_matmul_op.cc 75 out_shape.AddDim(lhs.dim_size(i));
89 out_shape.AddDim(lhs_rows);
90 out_shape.AddDim(rhs_cols);
nth_element_op.cc 69 out_shape.AddDim(input_in.dim_size(i));
dynamic_partition_op.cc 72 shape.AddDim(partition_count[p]);
74 shape.AddDim((*data)->dim_size(i));
lu_op.cc 82 batch_shape.AddDim(input.dim_size(dim));
94 permutation_shape.AddDim(num_rows);
lu_op_gpu.cu.cc 103 batch_shape.AddDim(input.dim_size(dim));
106 permutation_indices_shape.AddDim(num_rows);
mkl_reshape_op.cc 237 shape->AddDim(1);
242 shape->AddDim(size);
reduce_join_op.cc 104 if (keep_dims) output_shape.AddDim(1);
106 output_shape.AddDim(input_shape.dim_size(index));
padding_fifo_queue.cc 173 shape.AddDim(partial_shape.dim_size(j + 1));
180 shape.AddDim(max_val);
404 for (int64 s : partial.dim_sizes()) shape.AddDim(s < 0 ? 0 : s);
matrix_set_diag_op.cc 67 expected_diag_shape.AddDim(min_dim);
mkl_softmax_op.cc 150 output_tf_shape.AddDim((dst_pd.get_size() / sizeof(T)));
matrix_solve_op.cc 253 transposed_rhs_shape.AddDim(nrhs);
254 transposed_rhs_shape.AddDim(n);
  /external/tensorflow/tensorflow/core/util/
saved_tensor_slice_util.cc 118 shape->AddDim(dim);
example_proto_fast_parsing.cc     [all...]
example_proto_helper.cc 310 output_shapes->indices_shape.AddDim(total_num_features);
311 output_shapes->indices_shape.AddDim(2);
312 output_shapes->values_shape.AddDim(total_num_features);
346 out_shape.AddDim(batch_size);
349 for (const int dim : shape.dim_sizes()) out_shape.AddDim(dim);
example_proto_helper.h 186 dense_shape.AddDim(dense_shapes[i].dim_size(d));
252 dense_shape.AddDim(dense_shapes[i].dim_size(d));
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
dynamic_stitch_op.cc 156 new_shape.AddDim(indices[input_num].shape().dimensions(0));
159 new_shape.AddDim(data0_shape.dim_size(d));
index_ops_cpu.cc 86 output_shape.AddDim(input_shape.dim_size((d < axis) ? d : d + 1));
tensor_array_ops.cc 80 ta_shape.AddDim(resource->max_array_size());
163 ta_shape.AddDim(size);
517 ta_shape.AddDim(resource->max_array_size());
  /external/tensorflow/tensorflow/core/kernels/data/
concatenate_dataset_op.cc 199 output_tensorshape.AddDim(dims1[d]);
201 output_tensorshape.AddDim(-1);
padded_batch_dataset_op.cc 280 batch_component_shape.AddDim(0);
282 batch_component_shape.AddDim(padded_shape.dim_size(dim));
331 component_shape.AddDim(batch_component_shape.dim_size(i));
  /external/tensorflow/tensorflow/core/ops/
parsing_ops_test.cc 75 shapes[i].AddDim(-1);
80 shapes[i].AddDim(i + 1);
  /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
model_ops.cc 196 output_shape.AddDim(num_data);
197 output_shape.AddDim(num_outputs);
224 output_paths_shape.AddDim(param_proto_.inference_tree_paths() ? num_data
296 output_shape.AddDim(num_data);
397 output_shape.AddDim(param_proto_.num_features());
  /external/tensorflow/tensorflow/lite/delegates/flex/
buffer_map.cc 151 shape.AddDim(tensor->dims->data[i]);

Completed in 432 milliseconds

12 3 4 5