Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h308 std::vector<llvm_ir::IrArray>* param_in_reduced_shape_arrays);
Dir_emitter_unnested.cc2276 std::vector<IrArray>* param_in_reduced_shape_arrays) { in ConstructInputReducedShapeAndCastInputIrArrayToShape() argument
2278 param_in_reduced_shape_arrays->reserve(num_params); in ConstructInputReducedShapeAndCastInputIrArrayToShape()
2283 param_in_reduced_shape_arrays->push_back(IrArray()); in ConstructInputReducedShapeAndCastInputIrArrayToShape()
2290 param_in_reduced_shape_arrays->push_back( in ConstructInputReducedShapeAndCastInputIrArrayToShape()
3105 std::vector<IrArray> param_in_reduced_shape_arrays; in EmitKernel() local
3111 &param_reduced_shapes, &param_in_reduced_shape_arrays); in EmitKernel()
3158 param_in_reduced_shape_arrays[id]; in EmitKernel()