OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:AddDim
(Results
26 - 50
of
112
) sorted by null
1
2
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
1
2
3
4
5