Searched defs:input_vec (Results 1 – 9 of 9) sorted by relevance
181 const auto input_vec = input_tensor->vec<tstring>(); in Compute() local256 const auto input_vec = input_tensor->vec<tstring>(); in Compute() local
409 const auto& input_vec = input_tensor->flat<tstring>(); in Compute() local
175 std::vector<string> input_vec; in FormatInputShapes() local
342 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local364 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local457 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local
121 OpKernelContext* ctx, const DataVec* input_vec) { in SetTrtEngineInputs()
449 uint32_t input_vec = input_cp / KNOB_SIMD16_WIDTH; in PaPatchListSingle() local453 uint32_t input_vec = input_cp / KNOB_SIMD_WIDTH; in PaPatchListSingle() local509 uint32_t input_vec = input_cp / KNOB_SIMD16_WIDTH; in PaPatchListTerm() local567 uint32_t input_vec = input_cp / KNOB_SIMD16_WIDTH; in PaPatchListTerm_simd16() local
2927 std::vector<HloInstruction*> input_vec = {input1, input2}; in TEST_P() local2978 std::vector<HloInstruction*> input_vec = {input1, input2}; in TEST_P() local
5639 const int8x16_t input_vec = vld1q_s8(input_data + i); variable5726 const uint8x16_t input_vec = vld1q_u8(input_data + i); variable5804 const int8x16_t input_vec = vld1q_s8(input_data + i); variable5881 const uint8x16_t input_vec = vld1q_u8(input_data + i); variable
2986 std::vector<float> input_vec(n_elements); in TEST_P() local3386 std::vector<Input> input_vec{a}; in TEST_P() local