/external/tensorflow/tensorflow/core/ops/ |
H A D | candidate_sampling_ops_test.cc | 35 .Input({"a", 0, DT_INT64}) 55 .Input({"a", 0, DT_INT64}) 56 .Input({"b", 0, DT_INT64})
|
H A D | sparse_ops_test.cc | 93 .Input({"split_dim", 0, DT_INT64}) 94 .Input({"indices", 1, DT_INT64}) 95 .Input({"values", 2, DT_INT64}) 96 .Input({"shape", 3, DT_INT64}) 169 .Input({"a_indices", 1, DT_INT64}) 170 .Input({"a_values", 2, DT_INT64}) 171 .Input({"a_shape", 3, DT_INT64}) 172 .Input({"b", 3, DT_INT64}) 258 for (int i = 0; i < n; ++i) src_list.emplace_back("a", 0, DT_INT64);
|
H A D | io_ops_test.cc | 30 .Input({{"c", 0, DT_FLOAT}, {"d", 0, DT_INT64}}) 31 .Attr("T", {DT_FLOAT, DT_INT64}) 51 .Input({{"d", 0, DT_FLOAT}, {"e", 0, DT_INT64}}) 52 .Attr("T", {DT_FLOAT, DT_INT64}) 89 .Attr("dtypes", {DT_FLOAT, DT_INT64})
|
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
H A D | model_ops_test.cc | 50 .Input("c", 2, DT_INT64) 52 .Input("e", 5, DT_INT64) 72 .Input("c", 2, DT_INT64) 74 .Input("e", 5, DT_INT64)
|
/external/tensorflow/tensorflow/core/kernels/ |
H A D | sparse_conditional_accumulator_op.cc | 76 DataTypeVector expected_inputs = {DT_STRING_REF, DT_INT64, DT_INT64}; 78 expected_inputs.push_back(DT_INT64); 108 {DT_INT64, accumulator->dtype(), DT_INT64}),
|
H A D | sparse_reduce_sum_op_test.cc | 36 .Input(FakeInput(DT_INT64)) 38 .Input(FakeInput(DT_INT64)) 78 .Input(FakeInput(DT_INT64)) 80 .Input(FakeInput(DT_INT64)) 109 Tensor expected_indices(allocator(), DT_INT64, local 118 Tensor expected_shape(allocator(), DT_INT64, {1}); local
|
H A D | sparse_add_op_test.cc | 43 .Input(FakeInput(DT_INT64)) 45 .Input(FakeInput(DT_INT64)) 46 .Input(FakeInput(DT_INT64)) 48 .Input(FakeInput(DT_INT64)) 81 Tensor expected_indices(allocator(), DT_INT64, indices_shape); local 89 Tensor expected_shape(allocator(), DT_INT64, local 128 Tensor expected_indices(allocator(), DT_INT64, \ 138 Tensor expected_shape(allocator(), DT_INT64, \ 188 Tensor expected_indices(allocator(), DT_INT64, TensorShape({0, 2})); \ 194 Tensor expected_shape(allocator(), DT_INT64, \ [all...] |
H A D | priority_queue_op.cc | 46 component_types_.insert(component_types_.begin(), DT_INT64);
|
H A D | sparse_dense_binary_op_shared_test.cc | 45 .Input(FakeInput(DT_INT64)) 47 .Input(FakeInput(DT_INT64)) 61 .Input(FakeInput(DT_INT64)) 63 .Input(FakeInput(DT_INT64)) 240 Tensor indices(DT_INT64, TensorShape({total_nnz, kNumDims})); 242 Tensor shape(DT_INT64, TensorShape({kNumDims}));
|
H A D | conditional_accumulator_base_op.cc | 36 OP_REQUIRES_OK(ctx, ctx->MatchSignature({DT_STRING_REF, DT_INT64}, {}));
|
H A D | conditional_accumulator_op.cc | 75 DataTypeVector expected_inputs = {DT_STRING_REF, DT_INT64};
|
/external/tensorflow/tensorflow/core/framework/ |
H A D | function_testlib.cc | 90 {{"two"}, "Const", {}, {{"value", kTwo}, {"dtype", DT_INT64}}}, 91 {{"scale"}, "Cast", {"two"}, {{"SrcT", DT_INT64}, {"DstT", "$T"}}}, 107 {{"two"}, "Const", {}, {{"value", kTwo}, {"dtype", DT_INT64}}}, 111 {{"SrcT", DT_INT64}, {"DstT", DT_INT32}}},
|
H A D | types.h | 262 ToSet(DT_COMPLEX64) | ToSet(DT_COMPLEX128) | ToSet(DT_INT64) | 273 ToSet(DT_FLOAT) | ToSet(DT_DOUBLE) | ToSet(DT_INT32) | ToSet(DT_INT64) | 282 ToSet(DT_FLOAT) | ToSet(DT_DOUBLE) | ToSet(DT_INT64) | ToSet(DT_INT32) | 296 ToSet(DT_FLOAT) | ToSet(DT_DOUBLE) | ToSet(DT_INT32) | ToSet(DT_INT64) | 307 ToSet(DT_FLOAT) | ToSet(DT_INT32) | ToSet(DT_INT64) | ToSet(DT_HALF); 311 ToSet(DT_FLOAT) | ToSet(DT_INT32) | ToSet(DT_INT64) | ToSet(DT_QINT8) | 321 ToSet(DT_FLOAT) | ToSet(DT_INT32) | ToSet(DT_INT64) | ToSet(DT_QINT8) | 392 MATCH_TYPE_AND_ENUM(int64, DT_INT64); 422 ToSet(DT_COMPLEX64) | ToSet(DT_COMPLEX128) | ToSet(DT_INT64) | 451 ToSet(DT_INT32) | ToSet(DT_UINT32) | ToSet(DT_INT64) | ToSe [all...] |
/external/tensorflow/tensorflow/compiler/tf2xla/ |
H A D | literal_util_test.cc | 40 LiteralToHostTensor(*int64_values_literal, DT_INT64, &host_tensor) 66 LiteralToHostTensor(*int32_values_literal, DT_INT64, &host_tensor)
|
H A D | type_util.cc | 37 case tensorflow::DT_INT64:
|
/external/tensorflow/tensorflow/core/profiler/internal/ |
H A D | tfprof_tensor.cc | 53 case DataType::DT_INT64: { 57 } else if (tensor_->dtype() == DataType::DT_INT64) {
|
/external/tensorflow/tensorflow/compiler/jit/ |
H A D | xla_cpu_device.cc | 54 {DT_INT32, DT_INT64, DT_FLOAT, DT_DOUBLE, DT_COMPLEX64, DT_BOOL}};
|
H A D | xla_gpu_device.cc | 59 {DT_INT32, DT_INT64, DT_FLOAT, DT_DOUBLE, DT_COMPLEX64, DT_BOOL}};
|
H A D | xla_interpreter_device.cc | 29 {DT_INT32, DT_FLOAT, DT_BOOL, DT_DOUBLE, DT_INT64}};
|
/external/tensorflow/tensorflow/core/util/sparse/ |
H A D | sparse_tensor_test.cc | 105 Tensor ix(DT_INT64, TensorShape({N, NDIM, 1})); 115 Tensor ix(DT_INT64, TensorShape({N, NDIM})); 125 Tensor ix(DT_INT64, TensorShape({N, NDIM})); 135 Tensor ix(DT_INT64, TensorShape({N, NDIM})); 144 Tensor ix(DT_INT64, TensorShape({N, NDIM})); 162 Tensor ix(DT_INT64, TensorShape({N, NDIM})); 208 Tensor ix(DT_INT64, TensorShape({N, NDIM})); 227 Tensor ix(DT_INT64, TensorShape({N, NDIM})); 252 Tensor ix(DT_INT64, TensorShape({N, NDIM})); 295 Tensor ix(DT_INT64, TensorShap [all...] |
/external/tensorflow/tensorflow/core/util/ |
H A D | example_proto_helper_test.cc | 41 int64_dense_config.dtype = DT_INT64; 43 int64_dense_config.default_value = Tensor(DT_INT64, TensorShape({1})); 66 int64_sparse_config.dtype = DT_INT64; 163 Tensor int64_dense_output(DT_INT64, TensorShape({1, 1})); 191 Tensor int64_dense_output(DT_INT64, TensorShape({1, 1}));
|
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/ |
H A D | batch_features_testutil.cc | 62 auto indices_tensor = Tensor(tensorflow::DT_INT64, {sparse_values_size, 2}); 74 auto shape_tensor = Tensor(tensorflow::DT_INT64, {2});
|
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
H A D | sparse_column_iterable_test.cc | 31 const auto indices = Tensor(DT_INT64, {0, 2});
|
/external/tensorflow/tensorflow/core/common_runtime/sycl/ |
H A D | sycl_util.h | 47 case DT_INT64:
|
/external/tensorflow/tensorflow/contrib/factorization/kernels/ |
H A D | clustering_ops_test.cc | 55 Tensor sample_size(DT_INT64, TensorShape({})); 56 Tensor seed(DT_INT64, TensorShape({})); 57 Tensor num_retries_per_sample(DT_INT64, TensorShape({})); 122 Tensor seed(DT_INT64, TensorShape({})); 180 Tensor top(DT_INT64, TensorShape({}));
|