Home
last modified time | relevance | path

Searched defs:input_vec (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dstring_split_op.cc181 const auto input_vec = input_tensor->vec<tstring>(); in Compute() local
256 const auto input_vec = input_tensor->vec<tstring>(); in Compute() local
Dunicode_ops.cc409 const auto& input_vec = input_tensor->flat<tstring>(); in Compute() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcollective_ops_test.cc318 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local
338 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local
411 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local
504 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_show_multi.cc175 std::vector<string> input_vec; in FormatInputShapes() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_engine_utils.cc139 const DataVec* input_vec) { in SetTrtEngineInputs()
/external/mesa3d/src/gallium/drivers/swr/rasterizer/core/
Dpa_avx.cpp449 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
509 uint32_t input_vec = input_cp / KNOB_SIMD16_WIDTH; in PaPatchListTerm() local
567 uint32_t input_vec = input_cp / KNOB_SIMD16_WIDTH; in PaPatchListTerm_simd16() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc2889 std::vector<HloInstruction*> input_vec = {input1, input2}; in TEST_P() local
2940 std::vector<HloInstruction*> input_vec = {input1, input2}; in TEST_P() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h6127 const int8x16_t input_vec = vld1q_s8(input_data + i); variable
6214 const uint8x16_t input_vec = vld1q_u8(input_data + i); variable
6292 const int8x16_t input_vec = vld1q_s8(input_data + i); variable
6369 const uint8x16_t input_vec = vld1q_u8(input_data + i); variable
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc2408 std::vector<float> input_vec(TrtTensorDimsNumElements(actual_output_dims) * in TEST_F() local