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
347 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local367 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local440 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local533 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local
175 std::vector<string> input_vec; in FormatInputShapes() local
102 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
2889 std::vector<HloInstruction*> input_vec = {input1, input2}; in TEST_P() local2940 std::vector<HloInstruction*> input_vec = {input1, input2}; in TEST_P() local
5689 const int8x16_t input_vec = vld1q_s8(input_data + i); variable5776 const uint8x16_t input_vec = vld1q_u8(input_data + i); variable5854 const int8x16_t input_vec = vld1q_s8(input_data + i); variable5931 const uint8x16_t input_vec = vld1q_u8(input_data + i); variable
2371 std::vector<float> input_vec(n_elements); in TEST_P() local2772 std::vector<Input> input_vec{a}; in TEST_P() local