Home
last modified time | relevance | path

Searched refs:curr_dim_size (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc3539 int64 curr_dim_size = input_shape.dimensions(input_dim); in GetReductionToVectorDimensions() local
3543 num_reduced_major *= curr_dim_size; in GetReductionToVectorDimensions()
3547 num_reduced_minor *= curr_dim_size; in GetReductionToVectorDimensions()
3549 num_kept *= curr_dim_size; in GetReductionToVectorDimensions()