| /external/tensorflow/tensorflow/compiler/xla/client/ |
| sharding_builder.h | 31 using TileAssignment = Array<int64>;
51 OpSharding Tile1D(const Shape& tile_shape, int64 num_tiles);
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
| batch_features.h | 33 explicit BatchFeatures(int64 batch_size) : batch_size_(batch_size) {}
48 Status GetFeatureColumnSizes(int64* const num_dense_float_features,
49 int64* const num_sparse_float_features,
50 int64* const num_sparse_int_features) const {
65 ExamplesIterable examples_iterable(int64 example_start,
66 int64 example_end) const {
75 int64 batch_size() const { return batch_size_; }
79 const int64 batch_size_;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| argmax_op_gpu.cu.cc | 28 template struct functor::ArgMax<GPUDevice, T, int64>; \
29 template struct functor::ArgMin<GPUDevice, T, int64>; \
|
| cwise_op_equal_to_2.cc | 25 REGISTER6(BinaryOp, CPU, "Equal", functor::equal_to, int32, int64, complex64,
28 REGISTER6(BinaryOp, GPU, "Equal", functor::equal_to, int8, int16, int64,
|
| cwise_op_mul_2.cc | 25 REGISTER6(BinaryOp, CPU, "Mul", functor::mul, int8, uint16, int16, int64,
28 REGISTER6(BinaryOp, GPU, "Mul", functor::mul, int8, uint16, int16, int64,
|
| cwise_op_not_equal_to_2.cc | 25 REGISTER6(BinaryOp, CPU, "NotEqual", functor::not_equal_to, int32, int64,
28 REGISTER6(BinaryOp, GPU, "NotEqual", functor::not_equal_to, int8, int16, int64,
|
| cwise_op_pow.cc | 21 REGISTER2(BinaryOp, CPU, "Pow", functor::safe_pow, int32, int64);
25 int64);
|
| cwise_op_round.cc | 20 int32, int64);
28 int32, int64);
|
| data_format_ops_gpu.cu.cc | 27 template struct functor::DataFormatDimMap<GPUDevice, int64>;
29 template struct functor::DataFormatVecPermute<GPUDevice, int64>;
|
| diag_op.h | 29 Status operator()(OpKernelContext* context, const int64 size, const T* in,
35 Status operator()(OpKernelContext* context, const int64 size, const T* in,
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| periodic_function.cc | 27 const int64 interval_micros,
30 interval_micros_([interval_micros]() -> int64 {
67 void PeriodicFunction::RunLoop(const int64 start) {
70 const int64 deadline = start + options_.startup_delay_micros;
76 const int64 begin = options_.env->NowMicros();
81 const int64 end =
82 std::max(static_cast<int64>(options_.env->NowMicros()), begin);
85 const int64 deadline = begin + interval_micros_;
|
| /external/tensorflow/tensorflow/core/util/ |
| env_var.h | 32 // Returns an int64 into "value" from the environmental variable "env_var_name".
34 // If the string cannot be parsed into int64, an error status is returned.
35 Status ReadInt64FromEnvVar(StringPiece env_var_name, int64 default_val,
36 int64* value);
|
| work_sharder.cc | 23 void Shard(int max_parallelism, thread::ThreadPool* workers, int64 total,
24 int64 cost_per_unit, std::function<void(int64, int64)> work) {
45 static const int64 kMinCostPerShard = 10000;
47 std::max<int>(1, std::min(static_cast<int64>(max_parallelism),
56 const int64 block_size = (total + num_shards - 1) / num_shards;
64 for (int64 start = block_size; start < total; start += block_size) {
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| group_iterator.cc | 24 int64 N = iter_->ix_.dim_size(0);
25 auto ix_t = iter_->ix_.template matrix<int64>();
55 std::vector<int64> Group::group() const {
56 std::vector<int64> g;
57 auto ix_t = iter_->ix_.template matrix<int64>();
64 TTypes<int64>::UnalignedConstMatrix Group::indices() const {
65 return TTypes<int64>::UnalignedConstMatrix(
66 &(iter_->ix_.matrix<int64>()(loc_, 0)), next_loc_ - loc_, iter_->dims_);
|
| /frameworks/native/services/surfaceflinger/TimeStats/timestatsproto/ |
| timestats.proto | 26 optional int64 stats_start = 1;
27 optional int64 stats_end = 2;
45 optional int64 stats_start = 3;
46 optional int64 stats_end = 4;
|
| /prebuilts/go/darwin-x86/test/fixedbugs/ |
| bug156.go | 9 func f(a int64) int64 {
10 const b int64 = 0;
20 bug156.go:7: constant 18446744073709551615 overflows int64
|
| /prebuilts/go/darwin-x86/test/fixedbugs/issue9355.dir/ |
| a.go | 4 a, b, c int64
|
| /prebuilts/go/linux-x86/test/fixedbugs/ |
| bug156.go | 9 func f(a int64) int64 {
10 const b int64 = 0;
20 bug156.go:7: constant 18446744073709551615 overflows int64
|
| /prebuilts/go/linux-x86/test/fixedbugs/issue9355.dir/ |
| a.go | 4 a, b, c int64
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| runtime_matmul.cc | 26 using tensorflow::int64;
31 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, int64 m,
32 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) {
36 int64 lhs_rows = m;
37 int64 lhs_cols = k;
42 int64 rhs_rows = k;
43 int64 rhs_cols = n;
69 float* lhs, float* rhs, int64 m, int64 n
[all...] |
| runtime_single_threaded_matmul.cc | 23 using tensorflow::int64;
28 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, int64 m,
29 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs) {
30 int64 lhs_rows = m;
31 int64 lhs_cols = k;
36 int64 rhs_rows = k;
37 int64 rhs_cols = n;
63 const void* run_options_ptr, float* out, float* lhs, float* rhs, int64 m,
64 int64 n, int64 k, int32 transpose_lhs, int32 transpose_rhs)
[all...] |
| /prebuilts/go/darwin-x86/misc/cgo/test/ |
| issue5603.go | 20 var x [5]int64
21 exp := int64(C.issue5603exp)
22 x[0] = int64(C.issue5603foo0())
23 x[1] = int64(C.issue5603foo1(nil))
24 x[2] = int64(C.issue5603foo2(nil, nil))
25 x[3] = int64(C.issue5603foo3(nil, nil, nil))
26 x[4] = int64(C.issue5603foo4(nil, nil, nil, nil))
|
| /prebuilts/go/linux-x86/misc/cgo/test/ |
| issue5603.go | 20 var x [5]int64
21 exp := int64(C.issue5603exp)
22 x[0] = int64(C.issue5603foo0())
23 x[1] = int64(C.issue5603foo1(nil))
24 x[2] = int64(C.issue5603foo2(nil, nil))
25 x[3] = int64(C.issue5603foo3(nil, nil, nil))
26 x[4] = int64(C.issue5603foo4(nil, nil, nil, nil))
|
| /external/stressapptest/src/ |
| queue.h | 74 int64 nextin_;
75 int64 nextout_;
76 int64 q_size_; // Size of the queue.
77 int64 pushed_; // Number of pages pushed, total.
78 int64 popped_; // Number of pages popped, total.
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_module_config.h | 74 void set_replica_count(int64 replica_count) {
77 int64 replica_count() const { return replica_count_; }
95 int64 intra_op_parallelism_threads() const {
111 int64 replica_count_ = 1;
115 int64 intra_op_parallelism_threads_ = -1;
|
