| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 65 input.shape().AsProto(instr.mutable_input()->mutable_tensor_shape()); in LogConvAutotuneResults()
67 filter.shape().AsProto(instr.mutable_filter()->mutable_tensor_shape()); in LogConvAutotuneResults()
69 output.shape().AsProto(instr.mutable_output()->mutable_tensor_shape()); in LogConvAutotuneResults()
88 input.shape().AsProto(instr.mutable_input()->mutable_tensor_shape()); in LogFusedConvAutotuneResults()
90 filter.shape().AsProto(instr.mutable_filter()->mutable_tensor_shape()); in LogFusedConvAutotuneResults()
92 output.shape().AsProto(instr.mutable_output()->mutable_tensor_shape()); in LogFusedConvAutotuneResults()
94 bias.shape().AsProto(instr.mutable_bias()->mutable_tensor_shape()); in LogFusedConvAutotuneResults()
97 side_input->shape().AsProto( in LogFusedConvAutotuneResults()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_handle.cc | 30 void ResourceHandle::AsProto(ResourceHandleProto* proto) const { in AsProto() function in tensorflow::ResourceHandle
48 AsProto(&proto); in SerializeAsString()
73 p[i].AsProto(&proto); in EncodeResourceHandleList()
|
| D | attr_value_util.cc | 473 value.AsProto(out->mutable_shape()); in SetAttrValue()
481 value.AsProto(out->mutable_shape()); in SetAttrValue()
487 v.AsProto(out->mutable_list()->add_shape()); in SetAttrValue()
502 v.AsProto(out->mutable_list()->add_shape()); in SetAttrValue()
|
| D | variant_op_copy_test.cc | 178 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()
|
| D | tensor_slice_test.cc | 71 s.AsProto(&proto); in TEST()
106 s.AsProto(&proto); in TEST()
|
| D | tensor_shape_test.cc | 290 void AsProto(TensorShapeProto* proto) const;
464 void TensorShapeOld::AsProto(TensorShapeProto* proto) const { in AsProto() function in tensorflow::__anone850e9110111::TensorShapeOld
525 s.AsProto(&sp); in TEST()
526 sold.AsProto(&spold); in TEST()
|
| D | resource_handle.h | 61 void AsProto(ResourceHandleProto* proto) const;
|
| D | tensor_slice.h | 112 void AsProto(TensorSliceProto* proto) const;
|
| D | tensor.cc | 351 data[i].AsProto(handles->Add()); in Fill()
892 shape_.AsProto(proto->mutable_tensor_shape()); in AsProtoField()
902 shape_.AsProto(proto->mutable_tensor_shape()); in AsProtoTensorContent()
1199 shape().AsProto(description->mutable_shape()); in FillDescription()
|
| D | tensor_slice.cc | 113 void TensorSlice::AsProto(TensorSliceProto* proto) const { in AsProto() function in tensorflow::TensorSlice
|
| D | tensor_shape.h | 251 void AsProto(TensorShapeProto* proto) const;
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_slice_writer.h | 132 shape.AsProto(ssm->mutable_shape()); in Add()
137 slice.AsProto(ssm->add_slice()); in Add()
144 slice.AsProto(ss->mutable_slice()); in Add()
|
| /external/tensorflow/tensorflow/contrib/periodic_resample/ops/ |
| D | array_ops_test.cc | 30 shape.AsProto(&shape_proto); in TEST()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_util_test.cc | 50 output_shapes[0].AsProto(&shape_proto); in TEST()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 320 input.shape().AsProto(instr.mutable_input()->mutable_tensor_shape()); in LogFusedConvAutotuneResults()
322 filter.shape().AsProto(instr.mutable_filter()->mutable_tensor_shape()); in LogFusedConvAutotuneResults()
324 output.shape().AsProto(instr.mutable_output()->mutable_tensor_shape()); in LogFusedConvAutotuneResults()
326 bias.shape().AsProto(instr.mutable_bias()->mutable_tensor_shape()); in LogFusedConvAutotuneResults()
329 side_input->shape().AsProto( in LogFusedConvAutotuneResults()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_worker.cc | 100 val.shape().AsProto(tensor_proto->mutable_tensor_shape()); in GrpcRecvTensorAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_tensor_coding.cc | 125 val.shape().AsProto(skeleton.mutable_tensor_shape()); in EncodeSkeleton()
|
| /external/tensorflow/tensorflow/cc/ops/ |
| D | const_op_test.cc | 107 TensorShape({2, 2}).AsProto(proto.mutable_tensor_shape()); in TEST()
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.cc | 423 val.shape().AsProto(entry->mutable_shape()); in Add()
478 full_tensor_shape.AsProto(full_entry->mutable_shape()); in AddSlice()
480 slice_spec.AsProto(slice_proto); in AddSlice()
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| D | mpi_rendezvous_mgr.cc | 235 val.shape().AsProto(mpi_send_call->mRes_.mutable_response() in AddRequest()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 146 t->shape().AsProto(proto); in TensorHandleShape()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops_test.cc | 146 TensorShape{}.AsProto(shape_proto); in TEST()
149 TensorShape{1, 2, 3, 4}.AsProto(shape_proto); in TEST()
1782 shape.AsProto(&shape_proto); in TEST()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder_test.cc | 291 shape_1d.AsProto(input.mutable_tensor_shape()); in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 143 tensor.shape().AsProto(proto->mutable_tensor_shape()); in SetProtoFromGPU()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 444 conn.inside_shape.AsProto(&out_shape); in CreateTRTNode()
454 conn.outside_shape.AsProto(&in_shape); in CreateTRTNode()
|