Searched refs:index_operand (Results 1 – 8 of 8) sorted by relevance
69 auto index_operand = dynamic_op->mutable_operand(index_operand_number); in Run() local70 if (ShapeUtil::IsScalar(index_operand->shape())) { in Run()74 TF_RET_CHECK(index_operand->shape().rank() == 1); in Run()75 auto index_element_type = index_operand->shape().element_type(); in Run()80 ShapeUtil::MakeShape(index_element_type, {1}), index_operand, {dim}, in Run()
164 uint32_t index_operand = in AdjustAccessedIndices() local166 uint32_t member = literal_indices ? index_operand in AdjustAccessedIndices()168 ->GetDef(index_operand) in AdjustAccessedIndices()189 context->get_def_use_mgr()->GetDef(index_operand); in AdjustAccessedIndices()
173 uint32_t index_operand = in MarkAccessedMembersAsUsed() local175 uint32_t member = literal_indices ? index_operand in MarkAccessedMembersAsUsed()177 ->GetDef(index_operand) in MarkAccessedMembersAsUsed()
1033 mlir::Value index_operand = adaptor.getOperands()[0]; in matchAndRewrite() local1035 if (index_operand.getType().isa<tfrt::fallback::TFTensorType>()) { in matchAndRewrite()1037 index_operand = in matchAndRewrite()1047 if (!index_operand.getType().isa<tfrt::corert::TensorHandleType>()) in matchAndRewrite()1052 op.getLoc(), rewriter.getI32Type(), index_operand); in matchAndRewrite()