Searched refs:DT_INT64 (Results 101 - 125 of 144) sorted by relevance
123456
/external/tensorflow/tensorflow/tools/graph_transforms/ |
H A D | sparsify_gather.cc | 308 } else if (axis_t.dtype() == DT_INT64) { 357 DataType key_dtype = DT_INT64;
|
H A D | strip_unused_nodes_test.cc | 264 EXPECT_EQ(DT_INT64, node_lookup["a_node"]->attr().at("dtype").type());
|
/external/tensorflow/tensorflow/contrib/summary/ |
H A D | summary_ops_test.py | 48 types_pb2.DT_INT64: np.int64,
|
/external/tensorflow/tensorflow/contrib/tensorboard/db/ |
H A D | summary_db_writer_test.cc | 33 Tensor t(DT_INT64, TensorShape({}));
|
/external/tensorflow/tensorflow/core/kernels/ |
H A D | scatter_op_test.cc | 96 MakeOp(DT_FLOAT_REF, DT_INT64);
|
H A D | unique_op.cc | 69 axis_tensor.dtype() == DT_INT64),
|
H A D | where_op.cc | 141 context, context->allocate_temp(DT_INT64, TensorShape({}), &num_true));
|
H A D | example_parsing_ops_test.cc | 65 DataType dtype = DT_INT64;
|
H A D | restore_v2_op_test.cc | 312 MakeRestoreOp(DT_INT64);
|
H A D | random_op.cc | 277 shape_t.dtype() == DataType::DT_INT64), 286 } else if (shape_t.dtype() == DataType::DT_INT64) {
|
/external/tensorflow/tensorflow/core/util/ |
H A D | strided_slice_op.cc | 239 } else if (strides_tensor.dtype() == DT_INT64) {
|
H A D | tensor_slice_reader.cc | 262 READER_COPY(DT_INT64);
|
/external/tensorflow/tensorflow/core/common_runtime/ |
H A D | shape_refiner.cc | 501 } else if (node->output_type(0) == DT_INT64) { 538 } else if (node->output_type(0) == DT_INT64) { 739 } else if (scalar.dtype() == DT_INT64) {
|
H A D | shape_refiner_test.cc | 470 DT_INT64); 473 auto attrs = ops::Shape::OutType(DT_INT64); 590 DT_INT64); 593 auto attrs = ops::Size::OutType(DT_INT64);
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
H A D | model_pruner_test.cc | 201 Output a = ops::Variable(s.WithOpName("a"), {}, DT_INT64);
|
/external/tensorflow/tensorflow/core/kernels/data/ |
H A D | padded_batch_dataset_op.cc | 153 Tensor t(DT_INT64, TensorShape({padded_shapes_[i].dims()}));
|
/external/tensorflow/tensorflow/core/ops/ |
H A D | math_grad.cc | 79 {{"two"}, "Cast", {"c"}, {{"SrcT", DT_INT64}, {"DstT", "$T"}}},
|
H A D | math_ops_test.cc | 247 handle_data[2]->at(1).second = DT_INT64;
|
/external/tensorflow/tensorflow/python/lib/core/ |
H A D | py_func.cc | 132 *tf = DT_INT64;
|
/external/tensorflow/tensorflow/compiler/tests/ |
H A D | randomized_tests.cc | 466 case DT_INT64: { 516 case DT_INT64: { 734 case DT_INT64: 862 case DT_INT64: 1274 // auto type = Choose<DataType>({DT_INT32, DT_INT64}); 1288 // auto type = Choose<DataType>({DT_INT32, DT_INT64}); 2035 auto type = Choose<DataType>({DT_INT32, DT_INT64}); 2411 // TODO(b/31741996): re-enable DT_INT64 when bug is fixed. 2412 // DataType tpaddings = Choose<DataType>({DT_INT32, DT_INT64}); 2463 if (type == DT_INT64) retur [all...] |
/external/tensorflow/tensorflow/contrib/mpi_collectives/kernels/ |
H A D | mpi_ops.cc | 480 } else if (dtype == DT_INT64) { 499 } else if (dtype == DT_INT64) {
|
/external/tensorflow/tensorflow/core/grappler/clusters/ |
H A D | single_machine_test.cc | 389 const DataType key_dtype = DT_INT64; 390 const DataType data_dtype = DT_INT64;
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | tensor_array_ops.cc | 463 Tensor lengths(DT_INT64, {ta_dims[0]});
|
/external/tensorflow/tensorflow/contrib/nccl/kernels/ |
H A D | nccl_manager.cc | 85 case DT_INT64:
|
/external/tensorflow/tensorflow/core/grappler/ |
H A D | grappler_item_builder.cc | 60 } else if (type == DT_INT64) {
|
Completed in 8327 milliseconds
123456