- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for has_input (0.23 sec)
-
tensorflow/compiler/jit/kernels/xla_ops.cc
} XlaMergeOp::XlaMergeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} void XlaMergeOp::Compute(OpKernelContext* ctx) { VLOG(3) << "XlaMergeOp " << def().name(); int i = 0; if (ctx->has_input(i) || ctx->has_input(++i)) { ctx->set_output(0, ctx->input(i)); } } REGISTER_KERNEL_BUILDER(Name("XlaLaunch").Device(DEVICE_CPU), XlaLocalLaunchOp);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
auto cast_input = cast_op.getX(); // Update input type. auto transform_result_type = RankedTensorType::get(new_shape, getElementTypeOrSelf(cast_input)); cast_input.setType(transform_result_type); auto block_arg = mlir::dyn_cast<mlir::BlockArgument>(cast_input); auto cast_op_input = dyn_cast_or_null<TF::CastOp>(cast_input.getDefiningOp()); while (block_arg || cast_op_input) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
subprojects/core/src/main/java/org/gradle/api/internal/tasks/DefaultTaskInputs.java
} } }); } } private static class HasInputsVisitor implements PropertyVisitor { private boolean hasInputs; public boolean hasInputs() { return hasInputs; } @Override public void visitInputFileProperty( String propertyName, boolean optional,
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Mon Dec 11 13:37:56 UTC 2023 - 12.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool_patterns.td
(MHLO_CompositeOp:$old_val (variadic $a_input), ConstantStrAttr<StrAttr, "aten.avg_pool2d.default">, $attrs, $_, $_), (TFL_TransposeOp (TFL_AveragePool2DOp /*input*/ (TFL_TransposeOp $a_input, (Arith_ConstantOp
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_partitioned_op_conversion.cc
void runOnOperation() override; }; template <typename T> LogicalResult ReplacePartitionedOp(IntegerAttr num_cores_per_replica, T op) { constexpr bool is_input = std::is_same_v<std::decay_t<T>, TF::TPUPartitionedInputOp>; static_assert( is_input || std::is_same_v<std::decay_t<T>, TF::TPUPartitionedOutputOp>, "operator should either be an input or output"); OpBuilder builder(op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.td
// doesn't support non-constant weights. def ConvertTFDepthwiseConv2DToXLAConvOp : Pat< (TF_CastOp:$conv (TF_DepthwiseConv2dNativeOp (TF_CastOp:$cast_input (TF_SubOp (TF_CastOp $input, $truncate1), $input_zp), $truncate2), (TF_CastOp (TF_CastOp (TF_IdentityOp $filter), $truncate3), $truncate4), $strides, $padding, $explicit_padding,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Dec 10 05:52:02 UTC 2023 - 21.1K bytes - Viewed (0) -
platforms/core-execution/workers/src/integTest/groovy/org/gradle/workers/internal/WorkerExecutorParametersKotlinIntegrationTest.groovy
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Thu Jan 11 20:22:02 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize_layout.cc
LogicalResult matchAndRewrite(stablehlo::PadOp pad_op, PatternRewriter& rewriter) const override { Value pad_input = pad_op.getOperand(); RankedTensorType pad_type = pad_op.getType().cast<RankedTensorType>(); auto transpose_op = pad_input.getDefiningOp<stablehlo::TransposeOp>(); if (!transpose_op || !transpose_op->hasOneUse()) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 21:59:06 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td
(TFL_SoftmaxOp (TFL_SubOp:$sub $input, (TFL_ReduceMaxOp:$max $max_input, (Arith_ConstantOp I32ElementsAttr: $axes), ConstBoolAttrTrue), TFL_AF_None), $beta), (TFL_SoftmaxOp $input, $beta), [(IsSame $input, $max_input), (AxesIsLastDimension $axes, $max_input), (HasOneUse $sub), (HasOneUse $max)]>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 20:31:41 UTC 2024 - 66.4K bytes - Viewed (0) -
platforms/core-configuration/model-core/src/main/java/org/gradle/model/internal/registry/DefaultModelRegistry.java
} @Override public boolean isAchieved() { for (ModelNodeInternal child : scopeNode.getLinks()) { if (!child.isAtLeast(Discovered) && !hasInputs(new NodeAtState(child.getPath(), Discovered))) { return false; } } return true; } @Override
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Fri May 24 15:40:00 UTC 2024 - 45.7K bytes - Viewed (0)