| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | saved_variable_loading_test.cc | 68 saved_variable.set_dtype(dtype); in TEST_P()
84 saved_variable.set_dtype(dtype); in TEST_P()
101 saved_variable.set_dtype(dtype); in TEST_P()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unique_op_test.cc | 42 tensor_proto.set_dtype(DT_INT32); in GetRandomInt32TensorProto()
55 tensor_proto.set_dtype(DT_INT32); in GetRandomInt32TensorProtoWithRepeat()
116 tensor_proto.set_dtype(DT_STRING); in GetRandomStringsTensorProto()
|
| D | cast_op.cc | 104 out->set_dtype(dst_dtype_); in Compute()
106 out->set_dtype(external_dst_dtype_); in Compute()
|
| D | remote_fused_graph_execute_op_test.cc | 149 shape_a.set_dtype(DT_FLOAT); in BuildRemoteFusedGraphExecuteInfo()
156 shape_a_plus_b.set_dtype(DT_FLOAT); in BuildRemoteFusedGraphExecuteInfo()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.cc | 649 set_dtype(type); in Tensor()
656 set_dtype(type); in Tensor()
766 set_dtype(type); in Tensor()
780 set_dtype(type); in Tensor()
870 ret.set_dtype(dtype()); in Slice()
894 ret.set_dtype(dtype()); in SubSlice()
932 set_dtype(proto.dtype()); in FromProto()
947 proto->set_dtype(dtype()); in AsProtoField()
955 proto->set_dtype(dtype()); in AsProtoTensorContent()
1262 description->set_dtype(dtype()); in FillDescription()
|
| D | variant_op_copy_test.cc | 176 variant_proto.set_dtype(DT_VARIANT); in TEST()
210 variant_proto.set_dtype(DT_VARIANT); in TEST()
253 variant_proto.set_dtype(DT_VARIANT); in TEST()
|
| D | resource_handle.cc | 38 dtype_and_shape->set_dtype(dtype_and_shape_pair.dtype); in AsProto()
|
| D | tensor.h | 661 void set_dtype(DataType t) { shape_.set_data_type(t); } in set_dtype() function
703 set_dtype(dt); in set_shape()
712 set_dtype(other_dtype); in CopyFromInternal()
1030 set_dtype(DataTypeToEnum<T>::value); in Tensor()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transfer_utils.cc | 74 tensor_shape_type.set_dtype(input.second.dtype()); in BuildRemoteFusedGraphExecuteInfo()
89 tensor_shape_type_proto.set_dtype(tensor_shape_type->first); in BuildRemoteFusedGraphExecuteInfo()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder_test.cc | 223 input.set_dtype(DT_FLOAT); in TEST_F()
233 input.set_dtype(DT_FLOAT); in TEST_F()
264 input.set_dtype(DT_FLOAT); in TEST_F()
290 input.set_dtype(DT_FLOAT); in TEST_F()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_mgr.cc | 139 out->set_dtype(in->dtype); in SerializeRemoteTensorHandle()
147 dtype_and_shape_proto->set_dtype(dtype_and_shape.dtype); in SerializeRemoteTensorHandle()
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_tensor.cc | 39 tfprof_tensor_pb_.set_dtype(tensor_->dtype()); in Build()
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | cost_utils.cc | 56 tensor_prop.set_dtype(data_type); in GetTensorProperties()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | constant_utils.cc | 51 repr.set_dtype(tensorflow::DT_COMPLEX64); in CreateConstOpWithSingleValue()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 107 output_tensor->set_dtype(DT_FLOAT); in ExportFloatArray()
128 output_tensor->set_dtype(DT_FLOAT); in ExportFloatArray()
232 tensor->set_dtype(DT_BOOL); in ConvertBoolTensorConst()
256 tensor->set_dtype(DT_INT32); in ConvertIntTensorConst()
280 tensor->set_dtype(DT_INT32); in CreateIntTensorConst()
305 tensor->set_dtype(DT_COMPLEX64); in ConvertComplex64TensorConst()
328 tensor->set_dtype(DT_INT32); in CreateMatrixShapeTensorConst()
346 tensor->set_dtype(DT_INT32); in CreateDummyConcatDimTensorConst()
361 tensor->set_dtype(DT_INT32); in CreateReshapeShapeTensorConst()
751 max_bounds_const_op_tensor->set_dtype(DT_FLOAT); in ConvertRelu1Operator()
[all …]
|
| /external/tensorflow/tensorflow/core/data/ |
| D | compression_utils.cc | 52 metadata->set_dtype(component.dtype()); in CompressElement()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-variable-scope.pbtxt | 82 name: "set_dtype"
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | shape_optimizer.cc | 179 t->set_dtype(dtype); in Optimize()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 74 input->set_dtype(DT_FLOAT); in DescribeMatrix()
101 input->set_dtype(dtype); in DescribeArbitraryRankInput()
113 output->set_dtype(dtype); in DescribeArbitraryRankOutput()
171 tensor->set_dtype(DT_FLOAT); in DescribeTensor1D()
183 tensor->set_dtype(DT_FLOAT); in DescribeTensor4D()
196 tensor->set_dtype(DT_FLOAT); in DescribeTensor5D()
401 temp_tensor_proto.set_dtype(dtype); in GetTensorProto()
489 tensor->set_dtype(DT_FLOAT); in DescribeFusedBatchNorm()
495 tensor->set_dtype(DT_FLOAT); in DescribeFusedBatchNorm()
|
| D | utils.cc | 54 input.set_dtype(DataType::DT_INVALID); in UnknownInput()
190 input.set_dtype(output.dtype()); in FindInputFeatures()
|
| D | analytical_cost_estimator.cc | 97 output_info->set_dtype(output.dtype()); in AddCostNode()
|
| /external/tensorflow/tensorflow/c/ |
| D | python_api.cc | 102 out_shape_and_type->set_dtype(p.dtype); in GetHandleShapeAndType()
|
| /external/tensorflow/tensorflow/cc/ops/ |
| D | const_op_test.cc | 106 proto.set_dtype(DT_DOUBLE); in TEST()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_tensor_coding.cc | 126 skeleton.set_dtype(val.dtype()); in EncodeSkeleton()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | tensor_coding.cc | 169 tensor_meta->set_dtype(static_cast<DataType>(static_cast<int>(v))); in ParseTensorSubmessage()
|