Searched refs:DT_INT64 (Results 1 - 25 of 144) sorted by relevance

123456

/external/tensorflow/tensorflow/core/ops/
H A Dcandidate_sampling_ops_test.cc35 .Input({"a", 0, DT_INT64})
55 .Input({"a", 0, DT_INT64})
56 .Input({"b", 0, DT_INT64})
H A Dsparse_ops_test.cc93 .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 Dio_ops_test.cc30 .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 Dmodel_ops_test.cc50 .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 Dsparse_conditional_accumulator_op.cc76 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 Dsparse_reduce_sum_op_test.cc36 .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 Dsparse_add_op_test.cc43 .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 Dpriority_queue_op.cc46 component_types_.insert(component_types_.begin(), DT_INT64);
H A Dsparse_dense_binary_op_shared_test.cc45 .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 Dconditional_accumulator_base_op.cc36 OP_REQUIRES_OK(ctx, ctx->MatchSignature({DT_STRING_REF, DT_INT64}, {}));
H A Dconditional_accumulator_op.cc75 DataTypeVector expected_inputs = {DT_STRING_REF, DT_INT64};
/external/tensorflow/tensorflow/core/framework/
H A Dfunction_testlib.cc90 {{"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 Dtypes.h262 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 Dliteral_util_test.cc40 LiteralToHostTensor(*int64_values_literal, DT_INT64, &host_tensor)
66 LiteralToHostTensor(*int32_values_literal, DT_INT64, &host_tensor)
H A Dtype_util.cc37 case tensorflow::DT_INT64:
/external/tensorflow/tensorflow/core/profiler/internal/
H A Dtfprof_tensor.cc53 case DataType::DT_INT64: {
57 } else if (tensor_->dtype() == DataType::DT_INT64) {
/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_cpu_device.cc54 {DT_INT32, DT_INT64, DT_FLOAT, DT_DOUBLE, DT_COMPLEX64, DT_BOOL}};
H A Dxla_gpu_device.cc59 {DT_INT32, DT_INT64, DT_FLOAT, DT_DOUBLE, DT_COMPLEX64, DT_BOOL}};
H A Dxla_interpreter_device.cc29 {DT_INT32, DT_FLOAT, DT_BOOL, DT_DOUBLE, DT_INT64}};
/external/tensorflow/tensorflow/core/util/sparse/
H A Dsparse_tensor_test.cc105 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 Dexample_proto_helper_test.cc41 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 Dbatch_features_testutil.cc62 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 Dsparse_column_iterable_test.cc31 const auto indices = Tensor(DT_INT64, {0, 2});
/external/tensorflow/tensorflow/core/common_runtime/sycl/
H A Dsycl_util.h47 case DT_INT64:
/external/tensorflow/tensorflow/contrib/factorization/kernels/
H A Dclustering_ops_test.cc55 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({}));

Completed in 1194 milliseconds

123456