- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 51 for Dadd (0.03 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc
// Then branch gets the data and calls the reduce_function. auto& then_branch = dataset_if.getThenBranch(); then_branch.push_back(new Block); builder.setInsertionPointToEnd(&then_branch.front()); // Add iterator operational data access inside if. auto get_value = builder.create<TF::OptionalGetValueOp>(loc, dataset_types, get_next.getResult());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc
AddNodeAttr("function", function, &def); for (Node* node : nodes_to_remove) { VLOG(2) << "Deleting node " << node->DebugString(); // Ensure that we do not attempt to add control edges to nodes that are // deleted. control_inputs.erase(node); control_outputs.erase(node); graph->RemoveNode(node); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/tests/device-transform-gpu.mlir
// CHECK: %[[VAL_3:.*]] = tfl.mul(%[[VAL_1]], %[[VAL_2]]) <{fused_activation_function = "NONE"}> : (tensor<3xf32>, tensor<f32>) -> tensor<3xf32> // CHECK: %[[VAL_4:.*]] = tfl.add(%[[VAL_0]], %[[VAL_3]]) <{fused_activation_function = "NONE"}> : (tensor<1x384x384x3xf32>, tensor<3xf32>) -> tensor<1x384x384x3xf32> // CHECK: return %[[VAL_4]] : tensor<1x384x384x3xf32> // CHECK: } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 15.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc
}); return false; } return true; } }; void FusedKernelMatcherPass::runOnOperation() { RewritePatternSet patterns(&getContext()); auto func = getOperation(); patterns.add<FuseConv2DBiasAdd, FuseMatMulBiasAdd>(&getContext()); (void)applyPatternsAndFoldGreedily(func, std::move(patterns)); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.td
def CreateTensorScatterNdOp : NativeCodeCall< "$_builder.create<TF::ScatterNdOp>($0.getLoc(), $0.getType(), $1, $2, $3)">; //===----------------------------------------------------------------------===// // Add op patterns. //===----------------------------------------------------------------------===// // We can lower AddOp to AddV2Op, except for string type. AddOp is defined for
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 04 13:30:42 UTC 2024 - 24.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.h
// Monitoring Sampler APIs. // These APIs de-templated monitoring Sampler for swig. typedef struct TFE_MonitoringSamplerCell TFE_MonitoringSamplerCell; // Atomically add the value of the cell. TF_CAPI_EXPORT extern void TFE_MonitoringSamplerCellAdd( TFE_MonitoringSamplerCell* cell, double value); // Retrieves the current value of the cell. The return value is a HistogramProto
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 39.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc
// Insert all ops including nested ops for checking outputs/side effects. cluster_op.walk( [&](Operation* op) { tail_outside_compiled_ops_set.insert(op); }); // Only add top level ops to output vector. tail_outside_compiled_ops->push_back(&cluster_op); } *cluster_results = cluster_results_set.takeVector(); }
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/mlir/tensorflow/ir/tf_saved_model.cc
return success(); } }; void SessionInitializerOp::getCanonicalizationPatterns( RewritePatternSet &results, MLIRContext *context) { results.add<OptimizeSessionInitializerPattern>(context); } SmallVector<StringRef, 2> GetSessionInitializerExportedName(ModuleOp op) { auto session_initializer_op = GetSessionInitializerOp(op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.td
"tf_device.cluster"() ({ %group = "tf.Const"() <{value = dense<[[0, 1]]> : tensor<1x2xi32>}> : () -> tensor<1x2xi32> %arg1_reduced = "tf.XlaAllReduce"(%arg1_id, %group) <{mode = "CrossReplica", reduce_op = "Add"}> : (tensor<f32>, tensor<1x2xi32>) -> tensor<f32> "tf.OpA"(%arg1_reduced) : (tensor<f32>) -> () tf_device.return }) : () -> () tf_device.return } return }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 19.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.cc
TypeID::get<TF::XlaSpmdFullToShardShapeOp>(), TypeID::get<TF::XlaSpmdShardToFullShapeOp>(), TypeID::get<TF::XlaSvdOp>(), }; // Add the ops from the TPUEmbeddingOpsRegistry. for (auto op_type_id : TF::TPUEmbeddingOpsRegistry::Global().GetOpsTypeIds()) { ops_set->insert(op_type_id); } return ops_set; }();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 21.7K bytes - Viewed (0)