Searched defs:dimension_index (Results 1 – 4 of 4) sorted by relevance
160 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
1666 size_t dimension_index = 0; in test_shader_compilation() local1925 size_t dimension_index, in recursively_initialise()2491 size_t dimension_index, in recursively_initialise()2684 …g ExpressionsAssignment1<API>::recursively_initialise(std::string var_type, size_t dimension_index, in recursively_initialise()3214 …xpressionsIndexingArray2<API>::recursively_initialise(std::string var_type, size_t dimension_index, in recursively_initialise()
3355 int64 dimension_index = 0; in HandlePad() local