| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 66 instr.mutable_input()->set_dtype(input.dtype()); in LogConvAutotuneResults()
68 instr.mutable_filter()->set_dtype(filter.dtype()); in LogConvAutotuneResults()
70 instr.mutable_output()->set_dtype(output.dtype()); in LogConvAutotuneResults()
89 instr.mutable_input()->set_dtype(input.dtype()); in LogFusedConvAutotuneResults()
91 instr.mutable_filter()->set_dtype(filter.dtype()); in LogFusedConvAutotuneResults()
93 instr.mutable_output()->set_dtype(output.dtype()); in LogFusedConvAutotuneResults()
95 instr.mutable_bias()->set_dtype(bias.dtype()); in LogFusedConvAutotuneResults()
99 instr.mutable_side_input()->set_dtype(side_input->dtype()); in LogFusedConvAutotuneResults()
|
| D | unique_op_test.cc | 40 tensor_proto.set_dtype(DT_INT32); in GetRandomInt32TensorProto()
53 tensor_proto.set_dtype(DT_INT32); in GetRandomInt32TensorProtoWithRepeat()
107 tensor_proto.set_dtype(DT_STRING); in GetRandomStringsTensorProto()
|
| D | cast_op.cc | 110 out->set_dtype(dst_dtype_); in Compute()
112 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 | 607 Tensor::Tensor(DataType type) : shape_({0}), buf_(nullptr) { set_dtype(type); } in Tensor()
611 set_dtype(type); in Tensor()
646 set_dtype(other_dtype); in CopyFromInternal()
735 set_dtype(type); in Tensor()
749 set_dtype(type); in Tensor()
816 ret.set_dtype(dtype()); in Slice()
840 ret.set_dtype(dtype()); in SubSlice()
878 set_dtype(proto.dtype()); in FromProto()
893 proto->set_dtype(dtype()); in AsProtoField()
901 proto->set_dtype(dtype()); in AsProtoTensorContent()
[all …]
|
| 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 | tensor.h | 596 void set_dtype(DataType t) { shape_.set_data_type(t); } in set_dtype() function
653 set_dtype(dt); in set_shape()
994 set_dtype(DataTypeToEnum<T>::value); in Tensor()
|
| /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/contrib/saved_model/cc/saved_model/ |
| D | signature_def_utils_test.cc | 106 (*signature_def_.mutable_inputs())[key].set_dtype(dtype); in SetInputDataTypeForKey()
110 (*signature_def_.mutable_outputs())[key].set_dtype(dtype); in SetOutputDataTypeForKey()
|
| /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()
|
| D | graph_transferer.cc | 156 graph_input_node_info.set_dtype(input_node_info.second.dtype()); in LoadGraphFromProto()
181 graph_output_node_info.set_dtype(data_types.at(port)); in LoadGraphFromProto()
498 const_node_info.set_dtype(const_tensor.dtype()); in RegisterConstantNode()
551 const_node_info.set_dtype(tensor.dtype()); in RegisterConstTensor()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 321 instr.mutable_input()->set_dtype(input.dtype()); in LogFusedConvAutotuneResults()
323 instr.mutable_filter()->set_dtype(filter.dtype()); in LogFusedConvAutotuneResults()
325 instr.mutable_output()->set_dtype(output.dtype()); in LogFusedConvAutotuneResults()
327 instr.mutable_bias()->set_dtype(bias.dtype()); in LogFusedConvAutotuneResults()
331 instr.mutable_side_input()->set_dtype(side_input->dtype()); in LogFusedConvAutotuneResults()
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_tensor.cc | 36 tfprof_tensor_pb_.set_dtype(tensor_->dtype()); in Build()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 64 input->set_dtype(DT_FLOAT); in DescribeMatrix()
91 input->set_dtype(dtype); in DescribeArbitraryRankInput()
103 output->set_dtype(dtype); in DescribeArbitraryRankOutput()
143 tensor->set_dtype(DT_FLOAT); in DescribeTensor1D()
155 tensor->set_dtype(DT_FLOAT); in DescribeTensor4D()
341 temp_tensor_proto.set_dtype(dtype); in GetTensorProto()
429 tensor->set_dtype(DT_FLOAT); in DescribeFusedBatchNorm()
435 tensor->set_dtype(DT_FLOAT); in DescribeFusedBatchNorm()
|
| D | utils.cc | 53 input.set_dtype(DataType::DT_INVALID); in UnknownInput()
189 input.set_dtype(output.dtype()); in FindInputFeatures()
|
| D | analytical_cost_estimator.cc | 95 output_info->set_dtype(output.dtype()); in AddCostNode()
|
| /external/tensorflow/tensorflow/contrib/session_bundle/ |
| D | bundle_shim.cc | 261 tensor_info.set_dtype(it_tensor_info->second); in AddInputToSignatureDef()
291 tensor_info.set_dtype(it_tensor_info->second); in AddOutputToSignatureDef()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 104 output_tensor->set_dtype(DT_FLOAT); in ExportFloatArray()
125 output_tensor->set_dtype(DT_FLOAT); in ExportFloatArray()
229 tensor->set_dtype(DT_BOOL); in ConvertBoolTensorConst()
253 tensor->set_dtype(DT_INT32); in ConvertIntTensorConst()
276 tensor->set_dtype(DT_INT32); in CreateIntTensorConst()
301 tensor->set_dtype(DT_COMPLEX64); in ConvertComplex64TensorConst()
324 tensor->set_dtype(DT_INT32); in CreateMatrixShapeTensorConst()
342 tensor->set_dtype(DT_INT32); in CreateDummyConcatDimTensorConst()
357 tensor->set_dtype(DT_INT32); in CreateReshapeShapeTensorConst()
746 max_bounds_const_op_tensor->set_dtype(DT_FLOAT); in ConvertRelu1Operator()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | shape_optimizer.cc | 120 t->set_dtype(dtype); in Optimize()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-variable-scope.pbtxt | 82 name: "set_dtype"
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_worker.cc | 99 tensor_proto->set_dtype(val.dtype()); in GrpcRecvTensorAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_tensor_coding.cc | 124 skeleton.set_dtype(val.dtype()); in EncodeSkeleton()
|
| /external/tensorflow/tensorflow/cc/ops/ |
| D | const_op_test.cc | 106 proto.set_dtype(DT_DOUBLE); in TEST()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | cpp_shape_inference.cc | 149 out_shape_and_type->set_dtype(p.dtype); in RunCppShapeInferenceImpl()
|
| /external/tensorflow/tensorflow/c/ |
| D | python_api.cc | 139 out_shape_and_type->set_dtype(p.dtype); in GetHandleShapeAndType()
|