- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 263 for created (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_annotate_dynamic_shape_inputs.cc
class TPUAnnotateDynamicShapeInputsPass : public impl::TPUAnnotateDynamicShapeInputsPassBase< TPUAnnotateDynamicShapeInputsPass> { void runOnOperation() override; }; // Finds op that created a given value. If the value is a BlockArgument, this // returns the owner of the Block. Operation* GetOpOfValue(Value value) { if (auto block_arg = mlir::dyn_cast<BlockArgument>(value))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/add_quantization_unit_loc.cc
return isa<NameLoc>(loc) || (isa<CallSiteLoc>(loc) && isa<NameLoc>(mlir::cast<CallSiteLoc>(loc).getCallee())); }); } // Finds the pattern of the location created by `ImporterBase::GetLocation` // in `tensorflow/compiler/mlir/tensorflow/translate/import_model.cc`. void FindQuantizationUnitsRecursively(Location loc, SmallVector<QuantizationUnit>& units) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lift_variables.cc
// Create tensor attribute for this variable. absl::StatusOr<ElementsAttr> tensor_attr_or = ConvertTensor(tensor, &builder); if (!tensor_attr_or.ok()) { return module.emitOpError() << "failed to convert tensor (name: " << name.str() << ")"; } ElementsAttr tensor_attr = tensor_attr_or.value(); builder.create<tf_saved_model::GlobalTensorOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7.9K 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/c/c_api.cc
"and will trigger an error in the future. Either don't modify " "nodes after running them or create a new session."); } } } namespace { // Helper method that creates a shape handle for a shape described by dims. tensorflow::shape_inference::ShapeHandle ShapeHandleFromDims( tensorflow::shape_inference::InferenceContext* ic, int num_dims,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K 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/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/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/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/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)