| /external/tensorflow/tensorflow/c/ |
| c_test_util.cc | 55 TF_AllocateTensor(TF_INT32, dims, num_dims, sizeof(int32_t) * num_values);
69 return TF_NewTensor(TF_INT32, nullptr, 0, values, num_bytes,
87 TF_SetAttrType(desc, "dtype", TF_INT32);
246 TF_SetAttrType(desc, "T", TF_INT32);
|
| while_loop_test.cc | 99 EXPECT_EQ(TF_INT32, TF_TensorType(out));
|
| c_api_test.cc | 274 EXPECT_EQ(TF_INT32, static_cast<TF_DataType>(tensorflow::DT_INT32));
448 EXPECT_EQ(TF_INT32, TF_OperationOutputType(TF_Output{feed, 0}));
481 EXPECT_EQ(TF_INT32, TF_OperationOutputType(TF_Output{add, 0}));
487 EXPECT_EQ(TF_INT32, TF_OperationInputType(TF_Input{add, 0}));
488 EXPECT_EQ(TF_INT32, TF_OperationInputType(TF_Input{add, 1}));
[all...] |
| c_api_function_test.cc | 103 EXPECT_EQ(TF_INT32, TF_TensorType(out));
126 EXPECT_EQ(TF_INT32, TF_TensorType(out));
[all...] |
| c_api.h | 101 TF_INT32 = 3, // Int32 tensors are always in 'host' memory.
[all...] |
| /external/tensorflow/tensorflow/python/lib/core/ |
| ndarray_tensor_bridge.cc | 121 case TF_INT32:
|
| ndarray_tensor.cc | 84 *out_tf_datatype = TF_INT32;
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| tensor_jni.cc | 51 case TF_INT32:
54 static_assert(sizeof(jint) == 4, "Java int not compatible with TF_INT32");
94 CASE(TF_INT32, jint, "intValue", "()I", Int);
138 CASE(TF_INT32, jint, Int);
173 CASE(TF_INT32, jint, Int);
550 DEFINE_GET_SCALAR_METHOD(jint, TF_INT32, Int);
|
| /external/tensorflow/tensorflow/c/eager/ |
| c_api_test.cc | 67 TF_INT32, &dims[0], sizeof(dims) / sizeof(int64_t), sizeof(data));
88 TFE_OpSetAttrType(op, "Tidx", TF_INT32);
635 TF_SetAttrType(arg_descr, "dtype", TF_INT32);
641 TF_SetAttrType(id_descr, "T", TF_INT32);
661 TF_AllocateTensor(TF_INT32, nullptr, 0, 1 * sizeof(tensorflow::int32));
697 TF_SetAttrType(arg_descr, "dtype", TF_INT32);
703 TF_SetAttrType(id_descr, "T", TF_INT32);
723 TF_AllocateTensor(TF_INT32, nullptr, 0, 1 * sizeof(tensorflow::int32));
[all...] |
| /external/tensorflow/tensorflow/go/ |
| tensor.go | 41 Int32 DataType = C.TF_INT32
224 {reflect.TypeOf(int32(0)), C.TF_INT32},
|
| /external/tensorflow/tensorflow/python/eager/ |
| pywrap_tensor.cc | 314 if (handle_dtype != TF_INT32) {
674 TF_INT32, &num_tensors_int, /*num_dims=*/1, /*len=*/4 * num_tensors_int));
|
| /external/tensorflow/tensorflow/contrib/tensorrt/segment/ |
| segment_test.cc | 73 TF_SetAttrType(desc, "dtype", TF_INT32);
|
