Searched defs:dimension_index (Results 1 – 4 of 4) sorted by relevance
158 const int dimension_index = num_splits_and_index.index(); in HandleTileShardedInputs() local467 const auto dimension_index = dimension_and_output_splits.index(); in ValidateAndGetTiledExecuteOutputShape() local
119 int64_t left_pad = 0, right_pad = 0, dimension_index, index_in_input; in GetFlatIndex() local
1668 size_t dimension_index = 0; in test_shader_compilation() local1927 size_t dimension_index, in recursively_initialise()2493 size_t dimension_index, in recursively_initialise()2686 …g ExpressionsAssignment1<API>::recursively_initialise(std::string var_type, size_t dimension_index, in recursively_initialise()3216 …xpressionsIndexingArray2<API>::recursively_initialise(std::string var_type, size_t dimension_index, in recursively_initialise()
3445 int64_t dimension_index = 0; in HandlePad() local