Home
last modified time | relevance | path

Searched refs:TF_INT32 (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/java/src/main/native/
Dtensor_jni.cc51 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()
550 DEFINE_GET_SCALAR_METHOD(jint, TF_INT32, Int);
/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc67 TF_INT32, &dims[0], sizeof(dims) / sizeof(int64_t), sizeof(data)); in TestAxisTensorHandle()
88 TFE_OpSetAttrType(op, "Tidx", TF_INT32); in MinOp()
635 TF_SetAttrType(arg_descr, "dtype", TF_INT32); in TEST()
641 TF_SetAttrType(id_descr, "T", TF_INT32); in TEST()
661 TF_AllocateTensor(TF_INT32, nullptr, 0, 1 * sizeof(tensorflow::int32)); in TEST()
697 TF_SetAttrType(arg_descr, "dtype", TF_INT32); in TEST()
703 TF_SetAttrType(id_descr, "T", TF_INT32); in TEST()
723 TF_AllocateTensor(TF_INT32, nullptr, 0, 1 * sizeof(tensorflow::int32)); in TEST()
/external/tensorflow/tensorflow/c/
Dc_test_util.cc55 TF_AllocateTensor(TF_INT32, dims, num_dims, sizeof(int32_t) * num_values); in Int32Tensor()
69 return TF_NewTensor(TF_INT32, nullptr, 0, values, num_bytes, in Int32Tensor()
87 TF_SetAttrType(desc, "dtype", TF_INT32); in PlaceholderHelper()
246 TF_SetAttrType(desc, "T", TF_INT32); in Split3Helper()
Dc_api_test.cc274 EXPECT_EQ(TF_INT32, static_cast<TF_DataType>(tensorflow::DT_INT32)); in TEST()
448 EXPECT_EQ(TF_INT32, TF_OperationOutputType(TF_Output{feed, 0})); in TEST()
481 EXPECT_EQ(TF_INT32, TF_OperationOutputType(TF_Output{add, 0})); in TEST()
487 EXPECT_EQ(TF_INT32, TF_OperationInputType(TF_Input{add, 0})); in TEST()
488 EXPECT_EQ(TF_INT32, TF_OperationInputType(TF_Input{add, 1})); in TEST()
946 EXPECT_EQ(TF_INT32, TF_TensorType(out)); in TEST()
963 EXPECT_EQ(TF_INT32, TF_TensorType(out)); in TEST()
1007 EXPECT_EQ(TF_INT32, TF_TensorType(out)); in RunMinTest()
Dwhile_loop_test.cc99 EXPECT_EQ(TF_INT32, TF_TensorType(out)); in ExpectOutputValue()
Dc_api.h101 TF_INT32 = 3, // Int32 tensors are always in 'host' memory. enumerator
Dc_api_function_test.cc103 EXPECT_EQ(TF_INT32, TF_TensorType(out)); in RunT()
126 EXPECT_EQ(TF_INT32, TF_TensorType(out)); in Run()
/external/tensorflow/tensorflow/python/lib/core/
Dndarray_tensor_bridge.cc121 case TF_INT32: in TF_DataType_to_PyArray_TYPE()
Dndarray_tensor.cc84 *out_tf_datatype = TF_INT32; in PyArray_TYPE_to_TF_DataType()
/external/tensorflow/tensorflow/go/
Dtensor.go41 Int32 DataType = C.TF_INT32
224 {reflect.TypeOf(int32(0)), C.TF_INT32},
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.cc314 if (handle_dtype != TF_INT32) { in EagerTensor_init()
674 TF_INT32, &num_tensors_int, /*num_dims=*/1, /*len=*/4 * num_tensors_int)); in TFE_Py_TensorShapeSlice()
/external/tensorflow/tensorflow/contrib/tensorrt/segment/
Dsegment_test.cc73 TF_SetAttrType(desc, "dtype", TF_INT32); in PlaceholderHelper()