OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:AddDim
(Results
1 - 25
of
112
) sorted by null
1
2
3
4
5
/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/
k_feature_gradient_op.cc
113
out_routes_shape.
AddDim
(num_data);
114
out_routes_shape.
AddDim
(num_nodes);
118
out_data_shape.
AddDim
(num_nodes);
119
out_data_shape.
AddDim
(num_features);
123
out_weights_shape.
AddDim
(num_data);
124
out_weights_shape.
AddDim
(num_nodes);
125
out_weights_shape.
AddDim
(num_features_per_node);
stochastic_hard_routing_gradient_op.cc
134
output_routing_shape.
AddDim
(num_data);
135
output_routing_shape.
AddDim
(num_nodes);
139
output_data_shape.
AddDim
(num_nodes);
140
output_data_shape.
AddDim
(num_features);
144
output_parameters_shape.
AddDim
(num_data);
145
output_parameters_shape.
AddDim
(num_nodes);
146
output_parameters_shape.
AddDim
(num_features);
150
output_bias_shape.
AddDim
(num_data);
hard_routing_function_op.cc
116
output_probability_shape.
AddDim
(num_data);
117
output_probability_shape.
AddDim
(tree_depth_);
121
output_path_shape.
AddDim
(num_data);
122
output_path_shape.
AddDim
(tree_depth_);
stochastic_hard_routing_function_op.cc
129
output_probability_shape.
AddDim
(num_data);
130
output_probability_shape.
AddDim
(tree_depth_);
134
output_path_shape.
AddDim
(num_data);
135
output_path_shape.
AddDim
(tree_depth_);
unpack_path_op.cc
86
output_shape.
AddDim
(num_data);
87
output_shape.
AddDim
(num_nodes);
routing_function_op.cc
107
output_shape.
AddDim
(num_data);
108
output_shape.
AddDim
(max_nodes_);
routing_gradient_op.cc
106
output_shape.
AddDim
(num_data);
107
output_shape.
AddDim
(max_nodes_);
k_feature_routing_function_op.cc
122
output_shape.
AddDim
(num_data);
123
output_shape.
AddDim
(max_nodes_);
/external/tensorflow/tensorflow/core/kernels/
batchtospace_op.cc
137
external_output_shape.
AddDim
(orig_input_batch_size / block_shape_product);
143
external_output_shape.
AddDim
(size);
145
internal_input_shape.
AddDim
(input_batch_size);
146
internal_output_shape.
AddDim
(input_batch_size / block_shape_product);
161
internal_input_shape.
AddDim
(input_size);
162
internal_output_shape.
AddDim
(cropped_size);
163
external_output_shape.
AddDim
(cropped_size);
170
external_output_shape.
AddDim
(size);
173
internal_input_shape.
AddDim
(depth);
174
internal_output_shape.
AddDim
(depth)
[
all
...]
spacetobatch_op.cc
134
external_output_shape.
AddDim
(orig_input_tensor.dim_size(0) *
141
external_output_shape.
AddDim
(size);
143
internal_input_shape.
AddDim
(input_batch_size);
144
internal_output_shape.
AddDim
(input_batch_size * block_shape_product);
162
internal_input_shape.
AddDim
(input_size);
164
internal_output_shape.
AddDim
(output_size);
165
external_output_shape.
AddDim
(output_size);
172
external_output_shape.
AddDim
(size);
175
internal_input_shape.
AddDim
(depth);
176
internal_output_shape.
AddDim
(depth)
[
all
...]
svd_op_gpu.cu.cc
113
u_shape.
AddDim
(p);
114
u_shape.
AddDim
(m);
245
input_shape.
AddDim
(n);
246
input_shape.
AddDim
(m);
315
shapeS.
AddDim
(p);
318
shapeU.
AddDim
(m);
319
shapeU.
AddDim
(m);
320
shapeV.
AddDim
(n);
321
shapeV.
AddDim
(n);
323
shapeU.
AddDim
(m)
[
all
...]
reduction_ops_common.cc
24
for (auto size : out_reshape_) shape.
AddDim
(size);
31
for (auto size : out_shape_) shape.
AddDim
(size);
39
shape.
AddDim
(data_reshape_[i]);
42
shape.
AddDim
(data_reshape_[i]);
decode_raw_op.cc
54
out_shape.
AddDim
(0);
66
out_shape.
AddDim
(added_dim);
reshape_op.h
109
shape->
AddDim
(1);
114
shape->
AddDim
(size);
gather_op.cc
93
result_shape.
AddDim
(params.dim_size(i));
98
result_shape.
AddDim
(params.dim_size(i));
reshape_util.cc
71
output_shape.
AddDim
(1);
77
output_shape.
AddDim
(size);
diag_op.cc
55
out_shape.
AddDim
(diagonal.dim_size(i));
58
out_shape.
AddDim
(diagonal.dim_size(i));
95
out_shape.
AddDim
(tensor.dim_size(i));
matrix_diag_op.cc
63
output_shape.
AddDim
(input_shape.dim_size(i));
67
output_shape.
AddDim
(min_dim);
104
output_shape.
AddDim
(k);
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
reshape_op.cc
62
shape.
AddDim
(1);
67
shape.
AddDim
(size);
matrix_set_diag_op.cc
50
expected_diag_shape.
AddDim
(min_dim);
/external/tensorflow/tensorflow/core/framework/
tensor_shape_test.cc
67
s.
AddDim
(100);
179
s.
AddDim
(1);
181
s.
AddDim
(100000);
184
s.
AddDim
(2);
186
s.
AddDim
(4);
188
s.
AddDim
(3);
228
/// then call `
AddDim
()` on.
243
void
AddDim
(int64 size);
386
AddDim
(d.size());
394
AddDim
(s)
[
all
...]
/external/tensorflow/tensorflow/core/util/
strided_slice_op.cc
266
processing_shape->
AddDim
(shrink_i ? 1 : -1);
353
processing_shape->
AddDim
(size_i);
355
processing_shape->
AddDim
(-1);
367
final_shape->
AddDim
(processing_shape->dim_size(gather_index));
369
final_shape->
AddDim
(1);
/external/tensorflow/tensorflow/contrib/periodic_resample/ops/
array_ops.cc
57
result_shape.
AddDim
(target_dimensions[i]);
/external/tensorflow/tensorflow/lite/tools/accuracy/
utils.cc
55
shape.
AddDim
(tflite_tensor.dims->data[i]);
/external/tensorflow/tensorflow/compiler/tf2xla/
xla_resource.cc
137
ta_shape.
AddDim
(max_array_size_);
145
ta_shape.
AddDim
(max_array_size_);
170
ta_shape.
AddDim
(max_array_size_);
Completed in 465 milliseconds
1
2
3
4
5