| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | util.cc | 70 return TF_INT32; in GetTensorFlowDataType()
106 case TF_INT32: in GetTensorFlowLiteType()
|
| D | util_test.cc | 115 EXPECT_EQ(TF_INT32, GetTensorFlowDataType(kTfLiteInt32)); in TEST()
132 EXPECT_EQ(kTfLiteInt32, GetTensorFlowLiteType(TF_INT32)); in TEST()
|
| /external/tensorflow/tensorflow/c/ |
| D | tf_datatype.h | 50 TF_INT32 = 3, // Int32 tensors are always in 'host' memory. enumerator
|
| D | c_api_experimental_test.cc | 208 TFE_OpSetAttrType(reshape_op, "Tshape", TF_INT32); in TEST_F()
219 TFE_OpSetAttrType(fill_op, "Tshape", TF_INT32); in TEST_F()
|
| D | c_test_util.cc | 71 TF_AllocateTensor(TF_INT32, dims, num_dims, sizeof(int32_t) * num_values); in Int32Tensor()
85 return TF_NewTensor(TF_INT32, nullptr, 0, values, num_bytes, in Int32Tensor()
294 TF_SetAttrType(desc, "T", TF_INT32); in Split3Helper()
|
| D | c_test_util.h | 54 TF_DataType dtype = TF_INT32,
|
| D | c_api_test.cc | 298 EXPECT_EQ(TF_INT32, static_cast<TF_DataType>(tensorflow::DT_INT32)); in TEST()
472 EXPECT_EQ(TF_INT32, TF_OperationOutputType(TF_Output{feed, 0})); in TEST()
505 EXPECT_EQ(TF_INT32, TF_OperationOutputType(TF_Output{add, 0})); in TEST()
511 EXPECT_EQ(TF_INT32, TF_OperationInputType(TF_Input{add, 0})); in TEST()
512 EXPECT_EQ(TF_INT32, TF_OperationInputType(TF_Input{add, 1})); in TEST()
1004 EXPECT_EQ(TF_INT32, TF_TensorType(out)); in TEST()
1021 EXPECT_EQ(TF_INT32, TF_TensorType(out)); in TEST()
1065 EXPECT_EQ(TF_INT32, TF_TensorType(out)); in RunMinTest()
|
| D | while_loop_test.cc | 98 EXPECT_EQ(TF_INT32, TF_TensorType(out)); in ExpectOutputValue()
|
| D | c_api_function_test.cc | 105 EXPECT_EQ(TF_INT32, TF_TensorType(out)); in RunT()
128 EXPECT_EQ(TF_INT32, TF_TensorType(out)); in Run()
1284 TF_SetAttrType(desc, "dtype", TF_INT32); in NodeWithAttrHelper()
|
| D | kernels_test.cc | 606 TF_KernelBuilder_TypeConstraint(builder, "T", TF_DataType::TF_INT32, status); in TEST()
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | tensor_jni.cc | 51 case TF_INT32: in elemByteSize()
94 CASE(TF_INT32, jint, "intValue", "()I", Int); in writeScalar()
138 CASE(TF_INT32, jint, Int); in write1DArray()
173 CASE(TF_INT32, jint, Int); in read1DArray()
506 DEFINE_GET_SCALAR_METHOD(jint, TF_INT32, Int);
|
| /external/tensorflow/tensorflow/cc/experimental/base/tests/ |
| D | tensor_types_test_util.h | 41 static constexpr TF_DataType kDType = TF_INT32;
|
| D | tensor_test.cc | 152 Tensor::FromBuffer(/*dtype=*/TF_INT32, /*shape=*/shape, in TEST()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test_util.cc | 59 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_INT32, nullptr, 0, status); in TestScalarTensorHandle()
139 TFE_AllocateHostTensor(ctx, TF_INT32, &dims[0], num_dims, status); in TestTensorHandleWithDimsInt()
307 TF_Tensor* t = TFE_AllocateHostTensor(ctx, TF_INT32, &dims[0], in TestAxisTensorHandle()
328 TFE_OpSetAttrType(op, "Tidx", TF_INT32); in MinOp()
|
| D | dlpack.cc | 86 case TF_DataType::TF_INT32: in GetDlDataType()
180 *tf_dtype = TF_DataType::TF_INT32; in TfDataTypeFormDlDataType()
|
| D | c_api_experimental_test.cc | 249 TF_SetAttrType(arg_descr, "dtype", TF_INT32); in TEST()
255 TF_SetAttrType(id_descr, "T", TF_INT32); in TEST()
281 TF_AllocateTensor(TF_INT32, nullptr, 0, 1 * sizeof(tensorflow::int32)); in TEST()
|
| D | gradients_util.cc | 74 TFE_AllocateHostTensor(ctx, TF_INT32, &dims[0], num_dims, status.get()); in TensorHandleWithDimsIntHelper()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | ndarray_tensor_bridge.cc | 122 case TF_INT32: in TF_DataType_to_PyArray_TYPE()
|
| D | ndarray_tensor.cc | 126 *out_tf_datatype = TF_INT32; in PyArray_TYPE_to_TF_DataType()
187 *out_tf_datatype = TF_INT32; in PyArray_TYPE_to_TF_DataType()
|
| /external/tensorflow/tensorflow/go/ |
| D | tensor.go | 48 Int32 DataType = C.TF_INT32
395 {reflect.TypeOf(int32(0)), C.TF_INT32},
|
| /external/llvm-project/llvm/include/llvm/Analysis/Utils/ |
| D | TFUtils.h | 252 M(int32_t, TF_INT32) \
|
| /external/tensorflow/tensorflow/python/client/ |
| D | tf_session_wrapper.cc | 1185 .value("TF_INT32", TF_INT32) in PYBIND11_MODULE()
|