| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | collective_ops_test.cc | 318 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local
319 absl::c_iota(input_vec, 0); in XLA_TEST_F()
320 auto input_literal = LiteralUtil::CreateR1<float>(input_vec); in XLA_TEST_F()
338 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local
339 absl::c_iota(input_vec, 0); in XLA_TEST_F()
340 auto input_literal = LiteralUtil::CreateR1<float>(input_vec); in XLA_TEST_F()
411 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local
412 absl::c_iota(input_vec, 0); in XLA_TEST_F()
413 auto input_literal = LiteralUtil::CreateR1<float>(input_vec); in XLA_TEST_F()
504 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | string_split_op.cc | 181 const auto input_vec = input_tensor->vec<tstring>(); in Compute() local
182 const int64 batch_size = input_vec.dimension(0); in Compute()
203 skip_empty_ ? Split(input_vec(i), delimiter, str_util::SkipEmpty()) in Compute()
204 : Split(input_vec(i), delimiter, str_util::AllowEmpty()); in Compute()
256 const auto input_vec = input_tensor->vec<tstring>(); in Compute() local
257 const int64 batch_size = input_vec.dimension(0); in Compute()
275 std::vector<StringPiece> parts = SplitV2(input_vec(i), sep, maxsplit_); in Compute()
|
| D | unicode_ops.cc | 409 const auto& input_vec = input_tensor->flat<tstring>(); in Compute() local
430 for (int i = 0; i < input_vec.size(); ++i) { in Compute()
431 const string& input = input_vec(i); in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_engine_utils.cc | 139 const DataVec* input_vec) { in SetTrtEngineInputs() argument
140 int n_inputs = ctx ? ctx->num_inputs() : (input_vec ? input_vec->size() : 0); in SetTrtEngineInputs()
144 ctx ? StrCat(IONamePrefixes::kInputPHName, i) : input_vec->at(i).name; in SetTrtEngineInputs()
148 const Tensor& input_tensor = ctx ? ctx->input(i) : input_vec->at(i).tensor; in SetTrtEngineInputs()
|
| D | trt_engine_utils.h | 139 const DataVec* input_vec = nullptr);
|
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| D | tfprof_show_multi.cc | 175 std::vector<string> input_vec; in FormatInputShapes() local
178 input_vec.push_back(absl::StrFormat("%d:unknown", s.first)); in FormatInputShapes()
180 input_vec.push_back( in FormatInputShapes()
185 absl::StrFormat("input_type: %s", absl::StrJoin(input_vec, ",\t")); in FormatInputShapes()
|
| /external/mesa3d/src/gallium/drivers/swr/rasterizer/core/ |
| D | pa_avx.cpp | 449 uint32_t input_vec = input_cp / KNOB_SIMD16_WIDTH; in PaPatchListSingle() local
453 uint32_t input_vec = input_cp / KNOB_SIMD_WIDTH; in PaPatchListSingle() local
462 (const float*)(&PaGetSimdVector_simd16(pa, input_vec, slot)[i]); in PaPatchListSingle()
464 const float* pInputVec = (const float*)(&PaGetSimdVector(pa, input_vec, slot)[i]); in PaPatchListSingle()
509 uint32_t input_vec = input_cp / KNOB_SIMD16_WIDTH; in PaPatchListTerm() local
513 (const float*)(&PaGetSimdVector_simd16(pa, input_vec, slot)[i]); in PaPatchListTerm()
516 uint32_t input_vec = input_cp / KNOB_SIMD_WIDTH; in PaPatchListTerm()
519 const float* pInputVec = (const float*)(&PaGetSimdVector(pa, input_vec, slot)[i]); in PaPatchListTerm()
567 uint32_t input_vec = input_cp / KNOB_SIMD16_WIDTH; in PaPatchListTerm_simd16() local
570 const float* pInputVec = (const float*)(&PaGetSimdVector(pa, input_vec, slot)[i]); in PaPatchListTerm_simd16()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 247 input_vec = op.inputs[0]
249 input_rank = array_ops.rank(input_vec)
252 return gen_xla_ops.xla_dynamic_update_slice(array_ops.zeros_like(input_vec),
258 array_ops.shape(input_vec) - slice_size - begin_vec, shape)
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 6127 const int8x16_t input_vec = vld1q_s8(input_data + i); variable
6128 const int16x8_t first_half = vmovl_s8(vget_low_s8(input_vec));
6129 const int16x8_t second_half = vmovl_s8(vget_high_s8(input_vec));
6214 const uint8x16_t input_vec = vld1q_u8(input_data + i); variable
6215 const uint16x8_t first_half = vmovl_u8(vget_low_u8(input_vec));
6216 const uint16x8_t second_half = vmovl_u8(vget_high_u8(input_vec));
6292 const int8x16_t input_vec = vld1q_s8(input_data + i); variable
6293 const int16x8_t first_half = vmovl_s8(vget_low_s8(input_vec));
6294 const int16x8_t second_half = vmovl_s8(vget_high_s8(input_vec));
6369 const uint8x16_t input_vec = vld1q_u8(input_data + i); variable
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 2408 std::vector<float> input_vec(TrtTensorDimsNumElements(actual_output_dims) * in TEST_F() local
2410 std::iota(input_vec.begin(), input_vec.end(), 1); in TEST_F()
2411 const DataVec input_data{{"input", AsTensor<float>(input_vec)}}; in TEST_F()
2413 {"my_reshape", ConstructTensor<float>(input_vec.size())}}; in TEST_F()
2416 ElementsAreArray(input_vec)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_test.cc | 2889 std::vector<HloInstruction*> input_vec = {input1, input2}; in TEST_P() local
2908 shape, input_vec, init_vec, window, compute_tuple)); in TEST_P()
2940 std::vector<HloInstruction*> input_vec = {input1, input2}; in TEST_P() local
2959 shape, input_vec, init_vec, window, compute_tuple)); in TEST_P()
|