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
318 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local338 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local411 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local504 std::vector<float> input_vec(kNumElems); in XLA_TEST_F() local
175 std::vector<string> input_vec; in FormatInputShapes() local
139 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
6127 const int8x16_t input_vec = vld1q_s8(input_data + i); variable6214 const uint8x16_t input_vec = vld1q_u8(input_data + i); variable6292 const int8x16_t input_vec = vld1q_s8(input_data + i); variable6369 const uint8x16_t input_vec = vld1q_u8(input_data + i); variable
2408 std::vector<float> input_vec(TrtTensorDimsNumElements(actual_output_dims) * in TEST_F() local