| /external/tensorflow/tensorflow/compiler/xla/ |
| sparse_index_array.cc | 26 SparseIndexArray::SparseIndexArray(int64 max_indices, int64 rank,
27 std::vector<int64> indices)
35 SparseIndexArray::SparseIndexArray(int64 max_indices, int64 rank,
36 tensorflow::gtl::ArraySlice<int64> indices)
38 std::vector<int64>(indices.begin(), indices.end())) {}
40 SparseIndexArray::SparseIndexArray(int64 max_indices,
41 const Array2D<int64>& indices)
43 std::vector<int64>(indices.begin(), indices.end())) {
[all...] |
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| op_level_cost_estimator.h | 64 int64 batch; // Batch size.
65 int64 ix; // Input size x.
66 int64 iy; // Input size y.
67 int64 iz; // Input depth.
68 int64 kx; // Kernel x.
69 int64 ky; // Kernel y.
70 int64 oz; // Output depth.
71 int64 ox; // Output size x.
72 int64 oy; // Output size y.
73 int64 sx; // Stride x
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| deep_conv2d.cc | 48 static int64 GetDeepConvCost(int input_tile_rows, int input_tile_cols,
52 const int64 input_tile_spatial_size = input_tile_rows * input_tile_cols;
53 const int64 input_transform_cost =
57 const int64 product_cost = input_tile_spatial_size * in_depth * out_depth;
60 const int64 output_tile_spatial_size = out_tile_rows * out_tile_cols;
61 const int64 output_transform_cost =
65 const int64 row_tiles = (out_rows + out_tile_rows - 1) / out_tile_rows;
66 const int64 col_tiles = (out_cols + out_tile_cols - 1) / out_tile_cols;
67 const int64 num_tiles = row_tiles * col_tiles;
74 static int64 GetDirectConvCost(int filter_rows, int filter_cols, int in_depth
[all...] |
| cwise_op_gpu_abs.cu.cc | 22 DEFINE_UNARY6(abs, Eigen::half, float, double, int64, complex64, complex128);
|
| cwise_op_gpu_round.cu.cc | 22 DEFINE_UNARY5(round, Eigen::half, float, double, int32, int64);
|
| cwise_op_gpu_sign.cu.cc | 22 DEFINE_UNARY6(sign, Eigen::half, float, double, int64, complex64, complex128);
|
| cwise_op_gpu_sub.cu.cc | 22 DEFINE_BINARY6(sub, Eigen::half, float, double, int64, complex64, complex128);
|
| /external/libtextclassifier/util/calendar/ |
| calendar-icu.h | 34 int64 reference_time_ms_utc,
38 int64* interpreted_time_ms_utc) const;
|
| /external/libtextclassifier/util/strings/ |
| numbers.h | 40 // Like ParseInt32, but for int64.
41 bool ParseInt64(const char *c_str, int64 *value);
48 std::string IntToString(int64 input);
|
| /external/perf_data_converter/src/compat/ |
| int_compat.h | 35 using int64 = ::google::protobuf::int64;
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| index_ops_kernel_argmax_float_2d.cc | 32 int64* in_sizes = static_cast<int64*>(data[1]);
33 int64* out_sizes = static_cast<int64*>(data[2]);
39 int64* out_t = static_cast<int64*>(out);
41 TTypes<int64, 1>::Tensor out_eig(out_t, out_eig_sizes);
43 out_eig = in_eig.argmax(dim).cast<int64>();
|
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| range_coder_ops_util.h | 29 std::vector<int64>* merged_broadcast_shape_pointer,
30 std::vector<int64>* merged_storage_shape_pointer);
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| verbs_util.h | 27 static string AppendStepidToKey(const string& key, int64 step_id);
29 int64& step_id);
|
| /external/tensorflow/tensorflow/core/framework/ |
| bfloat16.h | 57 void FloatToBFloat16(const float* src, bfloat16* dst, int64 size);
58 void BFloat16ToFloat(const bfloat16* src, float* dst, int64 size);
|
| /external/tensorflow/tensorflow/core/lib/core/ |
| threadpool.h | 72 void ParallelFor(int64 total, int64 cost_per_unit,
73 std::function<void(int64, int64)> fn);
91 int64 total, int64 cost_per_unit,
92 const std::function<void(int64, int64, int)>& fn);
|
| /external/tensorflow/tensorflow/core/platform/cloud/ |
| time_util_test.cc | 23 int64 mtime_nsec;
30 int64 mtime_nsec;
|
| retrying_utils.h | 34 const int64 initial_delay_microseconds);
37 const int64 initial_delay_microseconds,
38 const std::function<void(int64)>& sleep_usec);
45 const int64 initial_delay_microseconds);
|
| time_util.h | 25 Status ParseRfc3339Time(const string& time, int64* mtime_nsec);
|
| /external/tensorflow/tensorflow/core/platform/ |
| variant_coding.h | 31 void EncodeVariantList(const Variant* variant_array, int64 n, string* out);
35 bool DecodeVariantList(const string& in, Variant* variant_array, int64 n);
|
| /external/tensorflow/tensorflow/stream_executor/ |
| trace_listener.h | 55 virtual void SynchronousMemcpyH2DBegin(int64 correlation_id,
56 const void* host_src, int64 size,
58 virtual void SynchronousMemcpyH2DComplete(int64 correlation_id,
61 virtual void SynchronousMemcpyD2HBegin(int64 correlation_id,
63 int64 size, void* host_dst) {}
64 virtual void SynchronousMemcpyD2HComplete(int64 correlation_id,
67 virtual void BlockHostUntilDoneBegin(int64 correlation_id, Stream* stream) {}
68 virtual void BlockHostUntilDoneComplete(int64 correlation_id,
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| exhaustive_f32_elementwise_op_test.cc | 25 public ::testing::WithParamInterface<std::pair<int64, int64>> {
32 std::pair<int64, int64> known_incorrect_range) {
33 int64 begin, end;
35 int64 input_size = end - begin;
42 for (int64 i = begin; i < end; i++) {
61 for (int64 i = 0; i < input_size; i++) {
74 // floats expressed as a zero extended int64):
75 std::pair<int64, int64> known_incorrect_range = {1, 8315654}
[all...] |
| /external/protobuf/src/google/protobuf/util/internal/ |
| constants.h | 53 const int64 kTimestampMinSeconds = -62135596800;
56 const int64 kTimestampMaxSeconds = 253402300799;
59 const int64 kDurationMinSeconds = -315576000000;
62 const int64 kDurationMaxSeconds = 315576000000;
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| padding.cc | 27 tensorflow::gtl::ArraySlice<int64> input_dimensions,
28 tensorflow::gtl::ArraySlice<int64> window_dimensions,
29 tensorflow::gtl::ArraySlice<int64> window_strides) {
42 std::vector<std::pair<int64, int64>> MakePadding(
43 tensorflow::gtl::ArraySlice<int64> input_dimensions,
44 tensorflow::gtl::ArraySlice<int64> window_dimensions,
45 tensorflow::gtl::ArraySlice<int64> window_strides, Padding padding) {
48 std::vector<std::pair<int64, int64>> low_high_padding
[all...] |
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| rendezvous_mgr_interface.h | 74 virtual RemoteRendezvous* Find(int64 step_id) = 0;
80 virtual void RecvLocalAsync(int64 step_id,
85 virtual Status RecvLocal(int64 step_id, const Rendezvous::ParsedKey& parsed,
92 virtual void Cleanup(int64 step_id) = 0;
|
| /external/tensorflow/tensorflow/core/grappler/ |
| devices.h | 34 int64 AvailableGPUMemory(int gpu_id);
|
