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

1 2 3 45

  /external/tensorflow/tensorflow/core/kernels/
depthwise_conv_grad_op.cc 588 input_shape.AddDim(in_sizes_data[i]);
    [all...]
mkl_fused_batch_norm_op.cc 668 tf_shape_dst.AddDim(dst_pd.get_size() / sizeof(T));
    [all...]
mkl_relu_op.cc 515 tf_shape_dst.AddDim(dst_pd.get_size() / sizeof(T));
684 tf_shape_diff_src.AddDim(diff_src_pd.get_size() / sizeof(T));
    [all...]
segment_reduction_ops.cc 487 output_shape.AddDim(output_rows);
489 output_shape.AddDim(data.dim_size(i));
    [all...]
conv_ops_fused_image_transform.cc     [all...]
encode_proto_op.cc 559 expected_sizes_shape.AddDim(field_descs_.size());
pooling_ops_3d.cc 507 output_shape.AddDim(shape_vec(i));
    [all...]
remote_fused_graph_execute_utils.cc 506 ts.AddDim(context->Value(dh));
    [all...]
nn_ops_test.cc 83 shape.AddDim(sizes.size());
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_compiler.cc 658 shape.AddDim(arg.max_array_size);
676 shape.AddDim(arg.max_array_size);
    [all...]
  /external/tensorflow/tensorflow/tools/benchmark/
benchmark_model.cc 545 input.shape.AddDim(sizes[i]);
  /external/tensorflow/tensorflow/core/ops/
array_ops.cc     [all...]

Completed in 707 milliseconds

1 2 3 45