OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:AddDim
(Results
51 - 75
of
112
) sorted by null
1
2
3
4
5
/external/tensorflow/tensorflow/core/kernels/
avgpooling_op.cc
262
output_shape.
AddDim
(shape_vec(i));
417
output_shape.
AddDim
(shape_vec(i));
496
output_shape.
AddDim
(shape_vec(i));
fft_ops.cc
160
temp_shape.
AddDim
(fft_shape[i - 1]);
191
full_fft_shape.
AddDim
(input_dims[0]);
195
full_fft_shape.
AddDim
(fft_shape[i - 1]);
mkl_concat_op.cc
125
output_shape.
AddDim
(output_concat_dim);
404
tf_shape_dst.
AddDim
((dst_pd.get_size() / sizeof(T)));
477
tf_shape_output.
AddDim
(dnn_shape_output.GetSerializeBufferSize());
determinant_op.cc
151
out_shape.
AddDim
(input.dim_size(dim));
291
out_shape.
AddDim
(input.dim_size(dim));
dynamic_stitch_op.cc
128
result_shape.
AddDim
(*first_dim_size);
130
result_shape.
AddDim
(data0.dim_size(d));
mirror_pad_op.cc
108
output_shape.
AddDim
(before + in0.dim_size(d) + after);
318
output_shape.
AddDim
(out_size);
argmax_op.cc
74
output_shape.
AddDim
(input_shape.dim_size((d < axis) ? d : d + 1));
dynamic_partition_op_test.cc
127
empty_one_dim.
AddDim
(0);
gather_nd_op.cc
132
result_shape.
AddDim
(params_shape.dim_size(i));
linalg_ops_common.cc
131
batch_shape->
AddDim
(in.dim_size(dim));
mkl_aggregate_ops.cc
231
output_tf_shape.
AddDim
((output_pd.get_size() / sizeof(T)));
mkl_avgpooling_op.cc
133
tensor_out_shape.
AddDim
(dnnLayoutGetMemorySize_F32(static_cast<dnnLayout_t>(
254
output_shape.
AddDim
(shape_vec(i));
324
tensor_out_shape.
AddDim
(dnnLayoutGetMemorySize_F32(static_cast<dnnLayout_t>(
592
orig_input_shape.
AddDim
(shape_vec(i));
mkl_lrn_op.cc
214
output_tf_shape.
AddDim
(num_bytes / sizeof(T));
248
workspace_tf_shape.
AddDim
(0);
281
workspace_tf_shape.
AddDim
(workspace_bytes);
464
output_tf_shape.
AddDim
(num_bytes / sizeof(T));
mkl_maxpooling_op.cc
115
tensor_out_shape.
AddDim
(dnnLayoutGetMemorySize_F32(static_cast<dnnLayout_t>(
126
workspace_shape.
AddDim
(dnnLayoutGetMemorySize_F32(static_cast<dnnLayout_t>(
287
mkl_input_backprop.
AddDim
(
666
workspace_tf_shape.
AddDim
(workspace_bytes);
[
all
...]
reduction_ops_common.h
118
for (auto s : data_reshape_) shape.
AddDim
(s);
dynamic_partition_op_gpu.cu.cc
219
shape.
AddDim
(e_part_count(p));
221
shape.
AddDim
(data->dim_size(i));
mkl_conv_grad_input_ops.cc
431
diff_src_tf_shape.
AddDim
(diff_src_pd.get_size() / sizeof(T));
553
output_tf_shape.
AddDim
(dst_pd.get_size() / sizeof(T));
tile_ops.cc
117
output_shape.
AddDim
(input.dim_size(i) * multiples_array[i]);
296
output_shape.
AddDim
(input.dim_size(i) / multiples_array[i]);
concat_op.cc
148
output_shape.
AddDim
(output_concat_dim);
edit_distance_op.cc
153
output_shape.
AddDim
(std::max(hypothesis_st_shape.dim_size(d),
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
tensor_list_ops.cc
67
list_shape.
AddDim
(leading_dim);
91
list_shape.
AddDim
(0);
stack_ops.cc
71
stack_shape.
AddDim
(resource->max_array_size());
/external/tensorflow/tensorflow/compiler/tf2xla/
shape_util.cc
96
tensor_shape->
AddDim
(shape.dimensions(i));
/external/tensorflow/tensorflow/python/lib/core/
py_seq_tensor.cc
156
shape->
AddDim
(length);
163
shape->
AddDim
(length);
/external/tensorflow/tensorflow/core/grappler/costs/
utils_test.cc
45
shape.
AddDim
(sizes.size());
Completed in 3281 milliseconds
1
2
3
4
5