Home
last modified time | relevance | path

Searched refs:num_elts (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dtensor.cc1058 string SummarizeArray(int64 limit, int64 num_elts, in SummarizeArray() argument
1070 if (num_elts > limit) strings::StrAppend(&ret, "..."); in SummarizeArray()
1081 if (num_elts > limit) strings::StrAppend(&ret, "..."); in SummarizeArray()
1089 const int64 num_elts = NumElements(); in SummarizeValue() local
1091 max_entries = num_elts; in SummarizeValue()
1093 size_t limit = std::min(max_entries, num_elts); in SummarizeValue()
1095 return strings::StrCat("uninitialized Tensor of ", num_elts, in SummarizeValue()
1101 return SummarizeArray<Eigen::half>(limit, num_elts, shape_, data, in SummarizeValue()
1105 return SummarizeArray<float>(limit, num_elts, shape_, data, print_v2); in SummarizeValue()
1108 return SummarizeArray<double>(limit, num_elts, shape_, data, print_v2); in SummarizeValue()
[all …]
/external/swiftshader/third_party/llvm-7.0/llvm/utils/
Dshuffle_select_fuzz_tester.py218 def get_random_type(ty, num_elts): argument
242 if num_elts == None:
243 num_elts = random.choice(range(2, 65))
254 return Type(is_float, width, num_elts)
350 assert args.num_elts == None or args.num_elts > 0, "num_elts must be a positive integer."
353 ty = get_random_type(args.type, args.num_elts)
/external/tensorflow/tensorflow/cc/framework/
Dcc_op_gen.cc149 string PrintArray(int64 num_elts, const T* array) { in PrintArray() argument
151 for (int64 i = 0; i < num_elts; ++i) { in PrintArray()
161 const int64 num_elts = t.NumElements(); in PrintTensor() local
164 return PrintArray(num_elts, t.flat<float>().data()); in PrintTensor()
166 return PrintArray(num_elts, t.flat<double>().data()); in PrintTensor()
168 return PrintArray(num_elts, t.flat<int32>().data()); in PrintTensor()
171 return PrintArray(num_elts, t.flat<uint8>().data()); in PrintTensor()
174 return PrintArray(num_elts, t.flat<uint16>().data()); in PrintTensor()
177 return PrintArray(num_elts, t.flat<int16>().data()); in PrintTensor()
180 return PrintArray(num_elts, t.flat<int8>().data()); in PrintTensor()
[all …]
/external/libchrome/base/containers/
Dcircular_deque_unittest.cc50 constexpr size_t num_elts = 9; in TEST() local
57 circular_deque<int> buf(num_elts); in TEST()
59 EXPECT_EQ(num_elts, buf.size()); in TEST()
60 EXPECT_EQ(num_elts, static_cast<size_t>(buf.end() - buf.begin())); in TEST()
62 for (size_t i = 0; i < num_elts; i++) in TEST()
69 circular_deque<int> buf(num_elts, value); in TEST()
71 EXPECT_EQ(num_elts, buf.size()); in TEST()
72 EXPECT_EQ(num_elts, static_cast<size_t>(buf.end() - buf.begin())); in TEST()
74 for (size_t i = 0; i < num_elts; i++) in TEST()
/external/swiftshader/third_party/llvm-7.0/llvm/test/CodeGen/NVPTX/
Dwmma.py12 num_elts = 4 if abcd in "cd" and itype == "f16" else 8;
13 return [elt_ty] * num_elts
/external/tensorflow/tensorflow/core/common_runtime/
Dbase_collective_executor.cc133 int64 num_elts = ChunkElts(i); in ChunkAlias() local
137 return (num_elts > 0) ? output_.Slice(start, start + num_elts) in ChunkAlias()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc85 const int64 num_elts = arg.shape().dim_size(0); in GetComputationInputs() local
86 for (int i = 0; i < num_elts; ++i) { in GetComputationInputs()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.cc305 int64 num_elts = num_bytes / DataTypeSize(*dtype); in AnalyzeInputs() local
306 VLOG(2) << "num_bytes " << num_bytes << " num_elts=" << num_elts; in AnalyzeInputs()
307 *sa_shape = TensorShape({num_elts}); in AnalyzeInputs()