- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 324 for created (0.22 sec)
-
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) -
tensorflow/compiler/jit/pjrt_base_device.cc
options.shape_determination_fns) { if (options.shape_determination_fns.empty()) { LOG(ERROR) << "shape_representation_fns must be non-empty."; } VLOG(1) << "Created PJRT base device " << options.compilation_device_name << " device_name: " << name(); } /*static*/ absl::StatusOr<const PjRtBaseDevice::Metadata*> PjRtBaseDevice::GetMetadataFromDevice(DeviceBase* device) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types.cc
!IsQintValueDefinedByIntToQintCast(op->getOperand(i))) { new_operands.push_back(rewriter.create<TF::CastOp>( op->getLoc(), orig_op_type, operands[i])); } else { new_operands.push_back(operands[i]); } } // Create a new UQ op. OperationState state(op->getLoc(), op->getName().getStringRef(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.cc
// Caches a XlaDeviceAllocator per <backend, device ordinal> pair. A // XlaDeviceAllocator is created on demand and is associated with a // XlaDevice. It outlives the device itself (for instance, the buffer // backing a tensor holds a pointer to the allocator for book-keeping, // and this buffer can outlast the device). class XlaDeviceAllocatorState { public: // Creates or returns a cached XlaDeviceAllocator for a given // backend and device_ordinal.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
llvm::SmallVector<mlir::TF::SplitOp, 4> split_ops_for_tiled_input; split_ops_for_tiled_input.reserve( input_sharding.tile_assignment_devices_size()); // Creates a tree of split nodes for sharding tiled inputs. Splits nodes // are created such that input data is sharded in row major order. // Split nodes at ith depth from the original input node represent nodes // that split the input data at i-th dimension.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_formation.cc
// Set insertion point to right after all operations in cluster. builder->setInsertionPoint(c.ops.back()->getNextNode()); // Create a stand-alone region to hold all instructions in the cluster. Region region; region.push_back(new Block); // Move all operations in cluster to newly created region, stripping their // "device" attribute since launch op already carries device information. Block* block = ®ion.front();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 05 13:30:21 UTC 2023 - 6.8K bytes - Viewed (0)