Home
last modified time | relevance | path

Searched refs:AsProto (Results 1 – 25 of 40) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Dresource_handle.cc30 void ResourceHandle::AsProto(ResourceHandleProto* proto) const { in AsProto() function in tensorflow::ResourceHandle
39 dtype_and_shape_pair.shape.AsProto(dtype_and_shape->mutable_shape()); in AsProto()
60 AsProto(&proto); in SerializeAsString()
85 p[i].AsProto(&proto); in EncodeResourceHandleList()
Dtensor_slice_test.cc71 s.AsProto(&proto); in TEST()
106 s.AsProto(&proto); in TEST()
Dvariant_op_copy_test.cc178 scalar_shape.AsProto(variant_proto.mutable_tensor_shape()); in TEST()
212 scalar_shape.AsProto(variant_proto.mutable_tensor_shape()); in TEST()
255 scalar_shape.AsProto(variant_proto.mutable_tensor_shape()); in TEST()
Dattr_value_util.cc514 value.AsProto(out->mutable_shape()); in SetAttrValue()
522 value.AsProto(out->mutable_shape()); in SetAttrValue()
528 v.AsProto(out->mutable_list()->add_shape()); in SetAttrValue()
543 v.AsProto(out->mutable_list()->add_shape()); in SetAttrValue()
Dresource_handle.h85 void AsProto(ResourceHandleProto* proto) const;
Dtensor_shape_test.cc313 void AsProto(TensorShapeProto* proto) const;
487 void TensorShapeOld::AsProto(TensorShapeProto* proto) const { in AsProto() function in tensorflow::__anon127dd62f0111::TensorShapeOld
548 s.AsProto(&sp); in TEST()
549 sold.AsProto(&spold); in TEST()
Dtensor_slice.h112 void AsProto(TensorSliceProto* proto) const;
Dtensor_slice.cc113 void TensorSlice::AsProto(TensorSliceProto* proto) const { in AsProto() function in tensorflow::TensorSlice
Dtensor.cc377 data[i].AsProto(handles->Add()); in Fill()
946 shape_.AsProto(proto->mutable_tensor_shape()); in AsProtoField()
956 shape_.AsProto(proto->mutable_tensor_shape()); in AsProtoTensorContent()
1263 shape().AsProto(description->mutable_shape()); in FillDescription()
/external/tensorflow/tensorflow/c/experimental/saved_model/core/
Dsaved_variable_loading_test.cc69 shape.AsProto(saved_variable.mutable_shape()); in TEST_P()
86 shape.AsProto(saved_variable.mutable_shape()); in TEST_P()
103 shape.AsProto(saved_variable.mutable_shape()); in TEST_P()
/external/tensorflow/tensorflow/core/util/
Dtensor_slice_writer.h131 shape.AsProto(ssm->mutable_shape()); in Add()
136 slice.AsProto(ssm->add_slice()); in Add()
143 slice.AsProto(ss->mutable_slice()); in Add()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util_test.cc50 output_shapes[0].AsProto(&shape_proto); in TEST()
/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_tensor_op.cc143 c->input(kValueInputIndex).shape().AsProto(&value_shape_proto); in CalculateOutputSize()
148 .AsProto(&default_value_shape_proto); in CalculateOutputSize()
159 partial_tensor_shape.AsProto(&shape_proto); in CalculateOutputSize()
Dtensor_list.cc49 element_shape.AsProto(&element_shape_proto); in Encode()
/external/tensorflow/tensorflow/core/data/
Dcompression_utils.cc53 component.shape().AsProto(metadata->mutable_tensor_shape()); in CompressElement()
/external/tensorflow/tensorflow/cc/ops/
Dconst_op_test.cc107 TensorShape({2, 2}).AsProto(proto.mutable_tensor_shape()); in TEST()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_tensor_coding.cc127 val.shape().AsProto(skeleton.mutable_tensor_shape()); in EncodeSkeleton()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_resource_ops.cc216 shape.AsProto(engine_instance.add_input_shapes()); in Compute()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_mgr.cc148 dtype_and_shape.shape.AsProto(dtype_and_shape_proto->mutable_shape()); in SerializeRemoteTensorHandle()
Deager_service_impl.cc162 t->shape().AsProto(proto); in TensorHandleShape()
166 shape.AsProto(proto); in TensorHandleShape()
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.cc450 val.shape().AsProto(entry->mutable_shape()); in Add()
505 full_tensor_shape.AsProto(full_entry->mutable_shape()); in AddSlice()
507 slice_spec.AsProto(slice_proto); in AddSlice()
/external/webrtc/audio/test/
Dlow_bandwidth_audio_test.py331 f.write(histograms.AsProto().SerializeToString())
/external/tensorflow/tensorflow/core/ops/
Darray_ops_test.cc179 TensorShape{}.AsProto(shape_proto); in TEST()
182 TensorShape{1, 2, 3, 4}.AsProto(shape_proto); in TEST()
1848 shape.AsProto(&shape_proto); in TEST()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc685 sizes_shape.AsProto(sizes_tensor_proto.mutable_tensor_shape()); in CreatePadNode()
741 shape.AsProto(tensor_proto.mutable_tensor_shape()); in CreateSplitNode()
921 shape.AsProto(tensor_proto.mutable_tensor_shape()); in CreateConcatNode()
973 begin_shape.AsProto(begin_tensor_proto.mutable_tensor_shape()); in CreateSliceNode()
991 sizes_shape.AsProto(sizes_tensor_proto.mutable_tensor_shape()); in CreateSliceNode()
2369 arg_shape.handle_shape.AsProto(arg->mutable_shape()); in BuildCompileNode()
2374 arg_shape.shape.AsProto(arg->mutable_shape()); in BuildCompileNode()
2721 shape_shape.AsProto(tensorshape_proto.mutable_tensor_shape()); in CreatePerHostDummyArgs()
2746 scalar_shape.AsProto(tensor_proto.mutable_tensor_shape()); in CreatePerHostDummyArgs()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.cc141 tensor.shape().AsProto(proto->mutable_tensor_shape()); in SetProtoFromGPU()

12