| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 48 ASSERT_EQ(TF_FLOAT, TF_TensorType(out)); in TestFakeIteratorStack()
96 ASSERT_EQ(TF_FLOAT, TF_TensorType(image)); in TEST()
238 TFE_OpSetAttrType(send_op, "T", TF_FLOAT); in TEST()
254 TFE_OpSetAttrType(recv_op, "tensor_type", TF_FLOAT); in TEST()
303 auto* sym_handle = TFE_NewTensorHandleFromTFOutput(node, TF_FLOAT); in TEST()
305 CHECK_EQ(TFE_TensorHandleDataType(sym_handle), TF_FLOAT); in TEST()
390 CHECK_EQ(TFE_TensorHandleDataType(retvals[0]), TF_FLOAT); in TEST_F()
457 TF_DataType types[kNumInputs] = {TF_FLOAT, TF_FLOAT, TF_FLOAT}; in TEST_F()
|
| D | c_api_test.cc | 133 TF_Tensor* t = TF_NewTensor(TF_FLOAT, dims, 2, values, num_bytes, in TEST()
136 EXPECT_EQ(TF_FLOAT, TF_TensorType(t)); in TEST()
153 TF_NewTensor(TF_FLOAT, nullptr, 0, nullptr, 0, &NoOpDeallocator, nullptr); in TEST()
160 TF_Tensor* t = TF_AllocateTensor(TF_FLOAT, dims, 2, num_bytes); in TEST()
161 EXPECT_EQ(TF_FLOAT, TF_TensorType(t)); in TEST()
177 TF_Tensor* t = TF_NewTensor(TF_FLOAT, dims, 2, values, num_bytes, in TEST()
293 EXPECT_EQ(TF_FLOAT, static_cast<TF_DataType>(tensorflow::DT_FLOAT)); in TEST()
1410 EXPECT_EQ(TF_FLOAT, TF_TensorType(out)); in TEST()
1712 TF_Tensor* t = TF_AllocateTensor(TF_FLOAT, dims, 2, sizeof(float) * 4); in FloatTensor2x2()
1723 TF_SetAttrType(desc, "dtype", TF_FLOAT); in FloatConst2x2()
[all …]
|
| D | kernels_test.cc | 45 EXPECT_EQ(TF_FLOAT, type); in MyCreateFunc()
|
| D | c_test_util.cc | 101 return TF_NewTensor(TF_FLOAT, nullptr, 0, values, num_bytes, in FloatTensor()
|
| D | c_api.h | 100 TF_FLOAT = 1, enumerator
|
| D | c_api_function_test.cc | 1564 RandomUniform(shape, TF_FLOAT, func_graph.get(), s.get()); in DefineStatefulFunction()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | util.cc | 60 return TF_FLOAT; in GetTensorFlowDataType()
62 return TF_FLOAT; in GetTensorFlowDataType()
84 case TF_FLOAT: in GetTensorFlowLiteType()
|
| D | util_test.cc | 110 EXPECT_EQ(TF_FLOAT, GetTensorFlowDataType(kTfLiteNoType)); in TEST()
111 EXPECT_EQ(TF_FLOAT, GetTensorFlowDataType(kTfLiteFloat32)); in TEST()
122 EXPECT_EQ(kTfLiteFloat32, GetTensorFlowLiteType(TF_FLOAT)); in TEST()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test_util.cc | 26 TF_Tensor* t = TF_AllocateTensor(TF_FLOAT, nullptr, 0, sizeof(float)); in TestScalarTensorHandle()
78 TF_FLOAT, &dims[0], sizeof(dims) / sizeof(int64_t), sizeof(data)); in TestMatrixTensorHandle()
92 TF_FLOAT, &dims[0], sizeof(dims) / sizeof(int64_t), sizeof(data)); in DoubleTestMatrixTensorHandle3X2()
106 TF_FLOAT, &dims[0], sizeof(dims) / sizeof(int64_t), sizeof(data)); in TestMatrixTensorHandle3X2()
|
| D | c_api_test.cc | 443 EXPECT_EQ(TF_FLOAT, TFE_TensorHandleDataType(h)); in TEST()
1426 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in CreateVariable()
1441 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in CreateVariable()
1446 TF_AllocateTensor(TF_FLOAT, nullptr, 0, sizeof(value)), TF_DeleteTensor); in CreateVariable()
1480 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in TEST()
1490 EXPECT_EQ(TF_FLOAT, TFE_TensorHandleDataType(value_handle)); in TEST()
1521 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in BM_ReadVariable()
1533 CHECK_EQ(TF_FLOAT, TFE_TensorHandleDataType(h)); in BM_ReadVariable()
1562 TF_AllocateTensor(TF_FLOAT, dims.data(), dims.size(), sizeof(float)); in TEST()
1584 TFE_OpSetAttrType(op, "T", TF_FLOAT); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | tensor_jni.cc | 50 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()
555 DEFINE_GET_SCALAR_METHOD(jfloat, TF_FLOAT, Float);
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | attr_builder_test.cc | 75 a.Set("T", TF_FLOAT); in TEST()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | ndarray_tensor_bridge.cc | 116 case TF_FLOAT: in TF_DataType_to_PyArray_TYPE()
|
| D | ndarray_tensor.cc | 78 *out_tf_datatype = TF_FLOAT; in PyArray_TYPE_to_TF_DataType()
426 TF_DataType dtype = TF_FLOAT; in PyArrayToTF_Tensor()
|
| /external/tensorflow/tensorflow/go/ |
| D | tensor.go | 39 Float DataType = C.TF_FLOAT
218 {reflect.TypeOf(float32(0)), C.TF_FLOAT},
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl_test.cc | 88 TF_FLOAT, &dims[0], sizeof(dims) / sizeof(int64_t), sizeof(data)); in SetTensorProto()
|