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

1 23 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 23 4 5