Searched defs:params_rank (Results 1 – 8 of 8) sorted by relevance
44 const auto params_rank = ValueRank(gather_op.params()); in ExpandOp() local183 const int params_rank = ValueRank(gather_op.params()); in ComputeLayoutForward() local255 const int params_rank = ValueRank(gather_op.params()); in ComputeLayoutBackward() local305 const absl::optional<Layout>& params_layout, int params_rank, in GatherNdGetOutputLayoutFromInput()339 Layout* params_layout, int params_rank, in GatherNdGetInputLayoutFromOutput()384 const auto params_rank = ValueRank(gather_op.params()); in ExpandOp() local487 const int params_rank = ValueRank(gather_op.params()); in ComputeLayoutForward() local522 const int params_rank = ValueRank(gather_op.params()); in ComputeLayoutBackward() local
71 const int params_rank = NumDimensions(params); in Prepare() local
3463 int params_rank = params_type.getShape().size(); in convertGatherOp() local3739 int params_rank = params_type.getShape().size(); in convertGatherNdOp() local
547 void Gather(int32_t params_rank, in Gather()
548 void Gather(int32_t params_rank, in Gather()
1336 int64_t params_rank = params_type.getRank(); in BuildGatherOp() local
1494 auto params_rank = params_ty.getRank(); in matchAndRewrite() local