- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 78 for getAxes (1.53 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.cc
if (constant_op.getValue().getNumElements() > 32) continue; while (!constant_op.getResult().hasOneUse()) { auto new_constant_op = builder.clone(*constant_op.getOperation()); constant_op.getResult().getUses().begin()->assign( dyn_cast<mlir::stablehlo::ConstantOp>(new_constant_op)); } } } void ReplaceStablehloOpsInMainFunctionWithXlaCallModuleOpsPass:: runOnOperation() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_island_coarsening.cc
bool result_captured = false; Value inner_op_result = std::get<0>(ret_vals); Value island_result = std::get<1>(ret_vals); for (auto& use : llvm::make_early_inc_range(island_result.getUses())) { if (islandBlocks.count(use.getOwner()->getBlock())) { // If the use is within our island group, forward the result from // inner op. use.set(inner_op_result);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_validate_inputs.cc
llvm::SmallVector<Operation*> GetSuccessors(Operation* op) { llvm::SmallVector<Operation*> successors; for (auto result : op->getParentOp()->getOpResults()) { for (auto& use : result.getUses()) { auto succ = use.getOwner(); successors.push_back(succ); } } return successors; } // Gets the predecessors of an op wrapped in tf_executor.island.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 06:51:01 UTC 2024 - 21.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
loc, DenseElementsAttr::get(tensorflow::GetTypeFromTFTensorShape( {}, rewriter.getIntegerType(64)), op.getAxis())); int64_t axis = op.getAxis(); Type prev_input_ty, inferred_ty; SmallVector<Value, 4> expanded_inputs; expanded_inputs.reserve(op.getN()); for (Value input : op.getValues()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
int32_t axis_value = op.getAxis(); if (axis_value < 0) axis_value += input_type.getRank() + 1; if (axis_value < 0 || axis_value >= input_type.getRank() + 1) return op.emitOpError() << "op attribute 'axis' should be in range [-rank - 1, rank + 1), " << "got rank = " << input_type.getRank() << ", and axis = " << op.getAxis(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc
for (const auto& idx_result : llvm::enumerate(island.getOutputs())) { Value result = idx_result.value(); bool has_external_use = false; for (OpOperand& use : llvm::make_early_inc_range(result.getUses())) { if (wrapped_ops.count(use.getOwner())) use.set(yield_op.getOperand(idx_result.index())); else has_external_use = true; } if (has_external_use) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 27.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
// with corresponding results of new parallel_execute op for (auto& use : llvm::make_early_inc_range( old_parallel_execute->getResult(output_index).getUses())) { use.set(new_parallel_execute->getResult(output_index)); } continue; } int tpu_cluster_output_index = output_index - num_results_pre_cluster; const auto& output_sharding =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0) -
pkg/kubelet/server/server_test.go
} return s, nil } func (fk *fakeKubelet) GetExec(_ context.Context, podFullName string, podUID types.UID, containerName string, cmd []string, streamOpts remotecommandserver.Options) (*url.URL, error) { if fk.getExecCheck != nil { fk.getExecCheck(podFullName, podUID, containerName, cmd, streamOpts) } // Always use testContainerID resp, err := fk.streamingRuntime.GetExec(&runtimeapi.ExecRequest{ ContainerId: testContainerID,
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Wed Apr 24 18:25:29 UTC 2024 - 51.5K bytes - Viewed (0) -
pkg/kubelet/kuberuntime/kuberuntime_container.go
return remote.ErrContainerStatusNil } return m.ReadLogs(ctx, status.GetLogPath(), containerID.ID, logOptions, stdout, stderr) } // GetExec gets the endpoint the runtime will serve the exec request from. func (m *kubeGenericRuntimeManager) GetExec(ctx context.Context, id kubecontainer.ContainerID, cmd []string, stdin, stdout, stderr, tty bool) (*url.URL, error) { req := &runtimeapi.ExecRequest{ ContainerId: id.ID,
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Tue Jun 04 06:25:43 UTC 2024 - 54.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
llvm::SmallVector<Value, 4> next_values_to_visit; for (Value value_to_visit : values_to_visit) { if (!visited_values.insert(value_to_visit).second) continue; for (auto& use : value_to_visit.getUses()) { Operation* owner = use.getOwner(); if (auto sharding = llvm::dyn_cast<mlir::TF::XlaShardingOp>(owner)) return sharding.get_XlaSharding();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 28.9K bytes - Viewed (0)