Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2962 int dim_id, in EmitBlock()
2965 if (block_sizes[dim_id] == 1) { in EmitBlock()
2968 llvm::Value* starting_tile_index_for_dim = starting_tile[dim_id]; in EmitBlock()
2970 index_typed_constant(block_sizes[dim_id]); in EmitBlock()
2974 index_typed_constant(dims_in_block[dim_id] - 1); in EmitBlock()
2976 dims_in_tile[dim_id] - in EmitBlock()
2977 (dims_in_block[dim_id] - 1) * block_sizes[dim_id]); in EmitBlock()
2986 block_dim_induction_var, dim_id, &b_); in EmitBlock()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1118 int dim_id) { in GetUnknownOutputDim() argument
1119 DimId id{node, index, dim_id}; in GetUnknownOutputDim()