HomeSort by relevance Sort by last modified time
    Searched refs:tensor_shape_proto (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/core/kernels/hexagon/
graph_transfer_utils.cc 75 TensorShapeProto& tensor_shape_proto = *tensor_shape_type.mutable_shape(); local
77 tensor_shape_proto.add_dim()->set_size(dim);
90 TensorShapeProto& tensor_shape_proto = local
93 tensor_shape_proto.add_dim()->set_size(dim);
hexagon_graph_execution_test.cc 220 TensorShapeProto& tensor_shape_proto = *tensor_shape_type.mutable_shape(); local
222 tensor_shape_proto.add_dim()->set_size(dim);
232 TensorShapeProto& tensor_shape_proto = *tensor_shape_type.mutable_shape(); local
234 tensor_shape_proto.add_dim()->set_size(dim);
  /external/tensorflow/tensorflow/core/framework/
tensor_util.h 280 TensorShapeProto tensor_shape_proto; local
281 internal::SetTensorProtoShape(shape, &tensor_shape_proto);
282 if (TensorShape(tensor_shape_proto).num_elements() != values.size()) {
289 tensor.mutable_tensor_shape()->Swap(&tensor_shape_proto);
  /external/tensorflow/tensorflow/core/grappler/costs/
op_level_cost_estimator_test.cc 937 TensorShapeProto tensor_shape_proto; local
    [all...]
op_level_cost_estimator.h 28 TensorShapeProto* tensor_shape_proto);
op_level_cost_estimator.cc     [all...]
  /external/tensorflow/tensorflow/core/kernels/
remote_fused_graph_execute_utils.cc 632 TensorShapeProto& tensor_shape_proto = *tensor_shape_type.mutable_shape(); local
634 tensor_shape_proto.add_dim()->set_size(dim);
657 TensorShapeProto& tensor_shape_proto = local
660 tensor_shape_proto.add_dim()->set_size(dim);
    [all...]

Completed in 407 milliseconds