- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 263 for created (0.11 sec)
-
tensorflow/compiler/jit/xla_launch_util.cc
if (pjrt_tensor_buffer != nullptr) { args->push_back(pjrt_tensor_buffer->pjrt_buffer()); } else { // Creates a PjRtBuffer from DeviceMemoryBase. The newly created // PjRtBuffer needs to be persisted till XLA execution is completed. auto dmem = se::DeviceMemoryBase( const_cast<char*>(tensor->tensor_data().data()),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/extract_tpu_copy_with_dynamic_shape_op.cc
auto new_launch_op = builder.create<tf_device::LaunchOp>( old_launch_op->getLoc(), old_launch_op->getDeviceAttr(), /*result_types=*/new_launch_op_results_types); new_launch_op.getBody().takeBody(old_launch_op->getBody()); new_launch_op.GetBody().getTerminator()->setOperands(new_launch_op_results); return new_launch_op; } // Create the new device launch op which wraps the copy op.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/convert_func_to_bfloat16.cc
new_results))) return failure(); // Update the regions. The dialect conversion framework wants new regions to // be created and updated, rather than updating the old op. Thus we use an // OperationState so we can add regions to the new op. OperationState state(op->getLoc(), op->getName().getStringRef(), operands,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
OpBuilder replacer(yield_op); island_op.getBody().push_back(new Block); replacer.create<mlir::func::ReturnOp>(yield_op.getLoc(), yield_op.getOperands()); yield_op.erase(); } // Remap the captured operands in the (former) island block with newly // created entry block arguments in the function body. {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc
} // Wraps block in a Launch. External uses of ops in the block will be return // values of the Launch and remapped to the Launch results. If `before` is set // to true, the Launch is created before `op`. Otherwise the Launch is created // after `op`. mlir::tf_device::LaunchOp CreateLaunchForBlock(OpBuilder* builder, Operation* op, bool before,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.6K bytes - Viewed (0) -
tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass_test.cc
Out(NodeWith( Op("Slice"), Name("begin/static_shaped_slice/static_shaped_slice"))), _))); } // New constants being created need to have control dependencies copied to // ensure correct control flow analysis in TF V2. TEST(SliceToDynamicSliceRewriteTest, WithControlDepsToConstant) { Scope root = Scope::NewRootScope()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 18.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
tensorflow::GetTypeFromTFTensorShape({1}, shape_dtype); auto expanded_start_index = rewriter->create<TF::ExpandDimsOp>( loc, vector_type, start_index, scalar_zero); auto start_position = rewriter->create<TF::ConcatOp>( loc, position_type, scalar_zero, ArrayRef<Value>({expanded_start_index, partial_start_position})); // Create the slice size tensor. This is done by concatenating `size` and // `partial_size`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
void MarkAsResult(AbstractTensorHandle* gradient) const override; void DeleteGradient(AbstractTensorHandle* gradient) const override; private: // The context where the aggregation op `Add` is to be created. AbstractContext* ctx_; }; // Returns the number of elements in the gradient tensor. int64_t TapeVSpace::NumElements(AbstractTensorHandle* tensor) const {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/functional_control_flow_to_regions.cc
} void runOnOperation() override; }; // Creates a call to function `func` in region `caller_region`. Use `args` as // the call arguments, and terminate the region with a yield. The arguments are // cast to the required type before the call. `use_region_args` control whether // the input arguments are used as is (for IfOp) or block arguments of the same // type as the input arguments are created and then used as call arguments (for // While).
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_tf_ops_pass.cc
builder.clone(*op, mapping); } // Creates the ReturnOp so that the per-host function returns the // correct values of the cloned operations. llvm::SmallVector<Value, 4> results_after_mapping; for (Value result : metadata.results) { results_after_mapping.push_back(mapping.lookupOrDefault(result)); } builder.create<func::ReturnOp>(loc, results_after_mapping);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.4K bytes - Viewed (0)