Searched defs:tensor_dimension (Results 1 – 2 of 2) sorted by relevance
255 const auto& tensor_dimension = dim_shard_and_index.value(); in VerifyPaddedDimensionNotSharded() local349 const auto& tensor_dimension = std::get<0>(tensor_dim_and_multiple); in VerifyTileOperandLayout() local490 const auto& tensor_dimension = std::get<1>(multiple_and_dim_sharding); in ComputeLayoutForward() local544 const auto& tensor_dimension = std::get<1>(multiple_and_dim_sharding); in ComputeLayoutBackward() local
334 FrameBuffer::Dimension tensor_dimension = {output_width_, output_height_}; in Postprocess() local