Searched refs:output_location (Results 1 – 5 of 5) 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()
145 for (size_t output_location : kernels_[i].output_locations[j]) { in Initialize() local146 input_alloc_attrs_[output_location] = in Initialize()274 for (size_t output_location : in RunAsync() local276 if (output_location >= input_start_index) { in RunAsync()279 inputs[output_location].Destroy(); in RunAsync()298 for (size_t output_location : kernel_state.output_locations[j]) { in RunAsync() local301 inputs[output_location].Init(*val.tensor); in RunAsync()
14 // "table(table_id).feature(feature_id).output_location" repeated fields within41 repeated OutputLocation output_location = 1; field
794 string output_location = output_directives_[i].output_location; in Run() local795 if (!HasSuffixString(output_location, ".zip") && in Run()796 !HasSuffixString(output_location, ".jar")) { in Run()797 AddTrailingSlash(&output_location); in Run()799 GeneratorContextImpl** map_slot = &output_directories[output_location]; in Run()1327 directive.output_location = value; in InterpretArgument()1330 directive.output_location = value.substr(colon_pos + 1); in InterpretArgument()
356 string output_location; member