Home
last modified time | relevance | path

Searched refs:TF_FLOAT (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/c/
Dkernels_test.cc73 EXPECT_EQ(TF_FLOAT, type); in MyCreateFunc()
455 EXPECT_EQ(TF_FLOAT, val); in TEST_F()
471 const TF_DataType list[] = {TF_FLOAT, TF_DOUBLE, TF_HALF, TF_COMPLEX128}; in TEST_F()
735 size_t tensor_size_bytes = TF_DataTypeSize(TF_FLOAT); in TEST_F()
737 /*context=*/ctx, /*index=*/0, /*dtype=*/TF_FLOAT, /*dims=*/&dim, in TEST_F()
739 validate_tensor(output, &dim, 1, TF_FLOAT); in TEST_F()
764 /*context=*/ctx, /*index=*/0, /*dtype=*/TF_FLOAT, /*dims=*/&dim, in TEST_F()
767 validate_tensor(output, &dim, 1, TF_FLOAT); in TEST_F()
787 size_t tensor_size_bytes = TF_DataTypeSize(TF_FLOAT) * 6; in TEST_F()
789 /*context=*/ctx, /*index=*/0, /*dtype=*/TF_FLOAT, /*dims=*/dim, in TEST_F()
[all …]
Dtf_datatype.h48 TF_FLOAT = 1, enumerator
Dc_api_experimental_test.cc207 TFE_OpSetAttrType(reshape_op, "T", TF_FLOAT); in TEST_F()
218 TFE_OpSetAttrType(fill_op, "T", TF_FLOAT); in TEST_F()
Dc_api_test.cc136 TF_Tensor* t = TF_NewTensor(TF_FLOAT, dims, 2, values, num_bytes, in TEST()
139 EXPECT_EQ(TF_FLOAT, TF_TensorType(t)); in TEST()
156 TF_NewTensor(TF_FLOAT, nullptr, 0, nullptr, 0, &NoOpDeallocator, nullptr); in TEST()
163 TF_Tensor* t = TF_AllocateTensor(TF_FLOAT, dims, 2, num_bytes); in TEST()
164 EXPECT_EQ(TF_FLOAT, TF_TensorType(t)); in TEST()
180 TF_Tensor* t = TF_NewTensor(TF_FLOAT, dims, 2, values, num_bytes, in TEST()
296 EXPECT_EQ(TF_FLOAT, static_cast<TF_DataType>(tensorflow::DT_FLOAT)); in TEST()
1448 EXPECT_EQ(TF_FLOAT, TF_TensorType(out)); in TEST()
1750 TF_Tensor* t = TF_AllocateTensor(TF_FLOAT, dims, 2, sizeof(float) * 4); in FloatTensor2x2()
1761 TF_SetAttrType(desc, "dtype", TF_FLOAT); in FloatConst2x2()
[all …]
Dc_test_util.cc101 return TF_NewTensor(TF_FLOAT, nullptr, 0, values, num_bytes, in FloatTensor()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dutil.cc60 return TF_FLOAT; in GetTensorFlowDataType()
62 return TF_FLOAT; in GetTensorFlowDataType()
98 case TF_FLOAT: in GetTensorFlowLiteType()
Dutil_test.cc110 EXPECT_EQ(TF_FLOAT, GetTensorFlowDataType(kTfLiteNoType)); in TEST()
111 EXPECT_EQ(TF_FLOAT, GetTensorFlowDataType(kTfLiteFloat32)); in TEST()
129 EXPECT_EQ(kTfLiteFloat32, GetTensorFlowLiteType(TF_FLOAT)); in TEST()
/external/tensorflow/tensorflow/c/eager/
Dc_api_test_util.cc34 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_FLOAT, nullptr, 0, status); in TestScalarTensorHandle()
98 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0], in TestMatrixTensorHandle()
113 TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0], num_dims, status); in TestMatrixTensorHandleWithInput()
126 TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0], num_dims, status); in TestTensorHandleWithDimsFloat()
156 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0], in TestMatrixTensorHandle100x100()
171 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0], in DoubleTestMatrixTensorHandle3X2()
185 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0], in TestMatrixTensorHandle3X2()
201 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in TestVariable()
220 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in TestVariable()
225 TF_AllocateTensor(TF_FLOAT, nullptr, 0, sizeof(value)), TF_DeleteTensor); in TestVariable()
Dcustom_device_test.cc116 TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT); in TEST()
135 TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT); in TEST()
150 TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT); in TEST()
201 TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT); in TEST()
220 TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT); in TEST()
236 TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT); in TEST()
Dc_api_unified_experimental_test.cc362 TF_AddFunctionParameter(graph_ctx, TF_FLOAT, {-1, nullptr}, status.get()); in TEST_P()
453 TF_AddFunctionParameter(graph_ctx, TF_FLOAT, {-1, nullptr}, status.get()); in TEST_P()
556 auto* arg0 = TF_AddFunctionParameter(graph_ctx, TF_FLOAT, {-1, nullptr}, s); in TEST_P()
558 auto* arg1 = TF_AddFunctionParameter(graph_ctx, TF_FLOAT, {-1, nullptr}, s); in TEST_P()
712 auto* arg0 = TF_AddFunctionParameter(graph_ctx, TF_FLOAT, {-1, nullptr}, s); in TEST_P()
714 auto* arg1 = TF_AddFunctionParameter(graph_ctx, TF_FLOAT, {-1, nullptr}, s); in TEST_P()
978 TF_AddFunctionParameter(graph_ctx, TF_FLOAT, {-1, nullptr}, status.get()); in TEST_P()
Ddlpack.cc80 case TF_DataType::TF_FLOAT: in GetDlDataType()
196 *tf_dtype = TF_DataType::TF_FLOAT; in TfDataTypeFormDlDataType()
Dc_api_test.cc185 EXPECT_EQ(TF_FLOAT, TFE_TensorHandleDataType(h)); in TEST()
1214 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in TEST()
1224 EXPECT_EQ(TF_FLOAT, TFE_TensorHandleDataType(value_handle)); in TEST()
1259 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in BM_ReadVariable()
1266 CHECK_EQ(TF_FLOAT, TFE_TensorHandleDataType(h)); in BM_ReadVariable()
1296 TF_AllocateTensor(TF_FLOAT, dims.data(), dims.size(), sizeof(float)); in TEST()
1318 TFE_OpSetAttrType(op, "T", TF_FLOAT); in TEST()
1355 TF_AllocateTensor(TF_FLOAT, dims.data(), dims.size(), sizeof(float)); in TEST()
1380 TFE_OpSetAttrType(op, "T", TF_FLOAT); in TEST()
1411 EXPECT_EQ(TF_FLOAT, TFE_TensorHandleDataType(h)); in TEST()
[all …]
Dgradients_util.cc46 TFE_AllocateHostTensor(ctx, TF_FLOAT, nullptr, 0, status.get()); in ScalarTensorHandleHelper()
60 TFE_AllocateHostTensor(ctx, TF_FLOAT, &dims[0], num_dims, status.get()); in TensorHandleWithDimsFloatHelper()
Dc_api_cluster_test.cc97 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in ReadVariable()
/external/tensorflow/tensorflow/java/src/main/native/
Dtensor_jni.cc50 case TF_FLOAT: in elemByteSize()
92 CASE(TF_FLOAT, jfloat, "floatValue", "()F", Float); in writeScalar()
136 CASE(TF_FLOAT, jfloat, Float); in write1DArray()
171 CASE(TF_FLOAT, jfloat, Float); in read1DArray()
504 DEFINE_GET_SCALAR_METHOD(jfloat, TF_FLOAT, Float);
/external/tensorflow/tensorflow/cc/experimental/base/tests/
Dtensor_types_test_util.h31 static constexpr TF_DataType kDType = TF_FLOAT;
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_testlib.cc137 TF_NewTensor(TF_FLOAT, nullptr, 0, values, num_bytes, &FloatDeallocator, in FloatTensorHandle()
151 TF_NewTensor(TF_FLOAT, &dims, 1 /* num_dims */, values, num_bytes, in VectorFloatTensorHandle()
223 Variable::Create(context, TF_FLOAT, /* Scalar */ {}, 0, device_name, in BasicTestsForTwoDevices()
Dparallel_device_lib_test.cc58 TFE_OpSetAttrType(handle_op.get(), "dtype", TF_FLOAT); in TEST()
76 TFE_OpSetAttrType(read_op.get(), "dtype", TF_FLOAT); in TEST()
114 TFE_OpSetAttrType(handle_op.get(), "dtype", TF_FLOAT); in TEST()
Dparallel_device_test.cc476 TF_SetAttrType(placeholder_desc, "dtype", TF_FLOAT); in RegisterCollectiveMulFunction()
483 TF_SetAttrType(reduce_desc, "T", TF_FLOAT); in RegisterCollectiveMulFunction()
/external/tensorflow/tensorflow/c/experimental/saved_model/internal/
Dsaved_model_api_test.cc231 EXPECT_EQ(TF_FLOAT, TF_TensorSpecDataType(tensor_spec_a)); in TEST_P()
240 EXPECT_EQ(TF_FLOAT, TF_TensorSpecDataType(tensor_spec_b)); in TEST_P()
253 EXPECT_EQ(TF_FLOAT, TF_TensorSpecDataType(tensor_spec_out)); in TEST_P()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dattr_builder_test.cc76 a.Set("T", TF_FLOAT); in TEST()
/external/tensorflow/tensorflow/python/lib/core/
Dndarray_tensor_bridge.cc116 case TF_FLOAT: in TF_DataType_to_PyArray_TYPE()
Dndarray_tensor.cc120 *out_tf_datatype = TF_FLOAT; in PyArray_TYPE_to_TF_DataType()
483 TF_DataType dtype = TF_FLOAT; in NdarrayToTensor()
/external/tensorflow/tensorflow/go/
Dtensor.go46 Float DataType = C.TF_FLOAT
393 {reflect.TypeOf(float32(0)), C.TF_FLOAT},
/external/llvm-project/llvm/include/llvm/Analysis/Utils/
DTFUtils.h246 M(float, TF_FLOAT) \

12