Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for replaceMap (0.19 sec)

  1. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

        }
        auto conv_op = static_cast<const ConcreteType *>(this)->createTFLOp(
            &state, rewriter, op->getLoc(), tf_op.getType(), input, filter, bias);
    
        rewriter.replaceOp(op, conv_op.getResult());
        return success();
      }
    
      const IntegerAttr intAttrOne;
    
     private:
      bool allow_bf16_and_f16_type_legalization_;
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc

                                                       operands[i], operands[i + 1])
                            : operands[i];
          }
          n = (n + 1) / 2;
        }
    
        rewriter.replaceOp(addn_op, operands[0]);
        return success();
      }
    };
    
    // Lowers DynamicStitch op with constant indices and with static input and
    // output shapes using Reshape, UnPack and Pack op.
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 74.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

        if ((ranked_type && ranked_type.getRank() == 0) ||
            !matchPattern(element_shape, m_Constant(&dense_elem_attr))) {
          // If no constant is spotted, just forward the operand.
          rewriter.replaceOp(op, {input});
          return success();
        }
    
        RankedTensorType shape_type =
            tensorflow::GetTypeFromTFTensorShape({-1}, rewriter.getIntegerType(32));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc

          }
          auto scast_op = rewriter.create<quantfork::StorageCastOp>(
              call_op.getLoc(), result_type, result);
          replace_map.insert(std::make_pair(result, scast_op));
        }
    
        for (auto replace_pair : replace_map) {
          Value result = replace_pair.first;
          quantfork::StorageCastOp scast_op = replace_pair.second;
          result.replaceAllUsesExcept(scast_op, scast_op);
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 54.5K bytes
    - Viewed (0)
Back to top