Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 9 of 9 for identity_op (0.22 sec)

  1. .bazelrc

    # https://github.com/tensorflow/tensorflow/issues/48919.
    # Users can still include debug info for a specific kernel, e.g. with:
    #     --config=dbg --per_file_copt=+tensorflow/core/kernels/identity_op.*@-g
    # Since this .bazelrc file is synced between the tensorflow/tensorflow repo and
    # the openxla/xla repo, also include debug info for files under xla/.
    build:dbg --per_file_copt=+.*,-tensorflow.*,-xla.*@-g0
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 17:12:54 UTC 2024
    - 52.9K bytes
    - Viewed (0)
  2. callbacks/preload.go

    			}
    
    			if results, ok := joinIdentityMap[utils.ToStringKey(fieldValues...)]; ok {
    				joinKey := utils.ToStringKey(joinFieldValues...)
    				identityMap[joinKey] = append(identityMap[joinKey], results...)
    			}
    		}
    
    		_, foreignValues = schema.GetIdentityFieldValuesMap(tx.Statement.Context, joinResults, joinRelForeignFields)
    	} else {
    		for _, ref := range rel.References {
    Registered: Wed Jun 12 16:27:09 UTC 2024
    - Last Modified: Wed Jun 12 10:52:33 UTC 2024
    - 11.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc

                                       val_bcast.getType(), block_arg))) {
        return failure();
      }
    
      OpBuilder before_cluster_builder(cluster);
      IdentityOp assigned_id = before_cluster_builder.create<IdentityOp>(
          val_bcast.getLoc(), block_arg.getType(), block_arg);
      assigned_id->setAttr(kICIWeightDistributionMlirBridgeMarker,
                           before_cluster_builder.getBoolAttr(true));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 13 18:52:07 UTC 2024
    - 13.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

    DatasetInput GetDatasetInput(Value value) {
      // TODO(haoliang): add an interface for DatasetOp to avoid the following
      // enumeration.
      // Iteratively tracing upwards if parent op is `IdentityOp` or `IdentityNOp`.
      while (
          llvm::isa_and_nonnull<IdentityOp, IdentityNOp>(value.getDefiningOp())) {
        value = value.getDefiningOp()->getOperand(
            mlir::cast<OpResult>(value).getResultNumber());
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/decompose_optionals.cc

        Value value = nullptr;
        for (auto v : op.getComponents()) {
          value = v;
        }
        if (!value) return failure();
        rewriter.replaceOpWithNewOp<TF::IdentityOp>(op, value.getType(), value);
        return success();
      }
    };
    
    class HandleOptionalGet : public OpRewritePattern<TF::OptionalGetValueOp> {
     public:
      using OpRewritePattern::OpRewritePattern;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 9.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

      }
    };
    
    struct ConvertIdentity : public OpConversionPattern<TF::IdentityOp> {
      using OpConversionPattern::OpConversionPattern;
    
      LogicalResult matchAndRewrite(
          TF::IdentityOp op, OpAdaptor adaptor,
          ConversionPatternRewriter &rewriter) const override {
        Value input = adaptor.getOperands()[0];
        rewriter.replaceOpWithNewOp<TF::IdentityOp>(
            op, input.getType(), adaptor.getOperands(), op->getAttrs());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td

              tf_executor.yield %identity : tensor<i32>
            }
            %island_1_result, %island_1_control = tf_executor.island {
              %identity_n:2 = "tf.IdentityN"(%arg1, %island_0_result) : (tensor<i32>, tensor<i32>) -> (tensor<i32>, tensor<i32>)
              tf_executor.yield %identity_n#0
            }
            tf_executor.fetch %island_0_result, %island_1_result : tensor<i32>, tensor<i32>
          }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 99.6K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

        }
        return success();
      }
    };
    
    // Bypasses IdentityN op.
    class ConvertIdentityNOp : public OpRewritePattern<TF::IdentityNOp> {
     public:
      using OpRewritePattern<TF::IdentityNOp>::OpRewritePattern;
      LogicalResult matchAndRewrite(TF::IdentityNOp op,
                                    PatternRewriter &rewriter) const override {
        rewriter.replaceOp(op, op.getOperands());
        return success();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

    such that dx = g(dy). In Python,
    
    ```python
    with tf.get_default_graph().gradient_override_map(
        {'IdentityN': 'OverrideGradientWithG'}):
      y, _ = identity_n([f(x), x])
    
    @tf.RegisterGradient('OverrideGradientWithG')
    def ApplyG(op, dy, _):
      return [None, g(dy)]  # Do not backprop to f(x).
    ```
      }];
    
      let arguments = (ins
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top