Searched refs:output_location (Results 1 – 6 of 6) sorted by relevance
50 TPUEmbeddingOutputLayout::OutputLocation* output_location = in AddDefaultEmbeddingOutputLayoutIfNeeded() local52 output_location->set_tensor_index(table_id); in AddDefaultEmbeddingOutputLayoutIfNeeded()53 output_location->set_dim0_offset(feature_index); in AddDefaultEmbeddingOutputLayoutIfNeeded()54 output_location->set_dim1_offset(0); in AddDefaultEmbeddingOutputLayoutIfNeeded()
186 for (size_t output_location : kernels_[i].output_locations[j]) { in Initialize() local187 input_alloc_attrs_[output_location] = in Initialize()330 for (size_t output_location : arg_output_locations_[j]) { in Run() local331 if (output_location >= input_start_index) { in Run()334 inputs[output_location].Destroy(); in Run()341 for (size_t output_location : in Run() local343 if (output_location >= input_start_index) { in Run()346 inputs[output_location].Destroy(); in Run()
14 // "table(table_id).feature(feature_id).output_location" repeated fields within41 repeated OutputLocation output_location = 1; field
5 dEQP-VK.draw.output_location.array.r8g8-uint-mediump-output-uint,Fail
894 std::string output_location = output_directives_[i].output_location; in Run() local895 if (!HasSuffixString(output_location, ".zip") && in Run()896 !HasSuffixString(output_location, ".jar")) { in Run()897 AddTrailingSlash(&output_location); in Run()899 GeneratorContextImpl** map_slot = &output_directories[output_location]; in Run()1729 directive.output_location = value; in InterpretArgument()1732 directive.output_location = value.substr(colon_pos + 1); in InterpretArgument()
409 std::string output_location; member