- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 184 for output_types (0.18 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
shape[split_dimension] = shape[split_dimension] / num_split; output_type = mlir::RankedTensorType::get(shape, input_type.getElementType()); } } else { output_type = input_type; } // Creates a split op that splits |src_input| along |split_dimension|. llvm::SmallVector<mlir::Type, 4> output_types(num_split, output_type); *split_op = builder->create<mlir::TF::SplitOp>(
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/tests/resource-alias-analysis-test.mlir
%iter_handle1 = "tf.IteratorV2"() {container = "c", shared_name = "v0", output_shapes = [#tf_type.shape<>], output_types = [!tf_res]} : () -> !tf_res // expected-remark@below {{Result #0, ID 1 : 0, 1}} %iter_handle2 = "tf.IteratorV2"() {container = "c", shared_name = "v1", output_shapes = [#tf_type.shape<>], output_types = [!tf_res]} : () -> !tf_res func.return } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jul 21 17:19:47 UTC 2023 - 22.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/functionalize-if-fail.mlir
func.func @main() { tf_executor.graph { %0 = tf_executor.island wraps "tf._TPUReplicate"() {computation = @foo, Tinputs = [], Tbroadcast_inputs = [], NumVariables = 0, Tguaranteed_constants = [], output_types = []} : () -> () loc("_TPUReplicate") tf_executor.fetch } func.return } func.func @foo() { tf_executor.graph {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize.cc
} // Collect all the quantized outputs and replace them by the results of // the new quantized op. llvm::SmallDenseMap<Value, int> outputs_replaced; SmallVector<Type, 4> output_types; output_types.reserve(quantizing_op->getNumResults()); for (const auto& enumerated_result : llvm::enumerate(quantizing_op->getResults())) { Value result = enumerated_result.value();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 23.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
arg_types.push_back(arg.getType()); arg_locs.push_back(arg.getLoc()); } SmallVector<Type> output_types; for (const Value output : xla_call_module_op.getOutput()) { output_types.push_back(output.getType()); } entry_func_op.setFunctionType( rewriter.getFunctionType(arg_types, output_types)); // Replace argument types and locs. Block& entry = entry_func_op->getRegion(0).front();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_data_optimization.td
def FuseMapAndBatch : Pat< (TF_BatchDatasetV2Op (TF_MapDatasetOp $input_dataset, $other_arguments, $f, $output_types, $output_shapes, $use_inter_op_parallelism, $preserve_cardinality, $force_synchronous, $map_dataset_metadata), $batch_size, $drop_remainder, $parallel_copy, $batch_output_types,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 1.7K bytes - Viewed (0) -
tensorflow/cc/framework/cc_op_gen.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 17:22:47 UTC 2023 - 17K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/functionalize-if.mlir
func.func @main() { tf_executor.graph { %0 = tf_executor.island wraps "tf._TPUReplicate"() {computation = @foo, Tinputs = [], Tbroadcast_inputs = [], NumVariables = 0, Tguaranteed_constants = [], output_types = []} : () -> () loc("_TPUReplicate") tf_executor.fetch } func.return } func.func @foo() { tf_executor.graph {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/analyze-variables.mlir
output_shapes = [#tf_type.shape<?>], output_types = [!tf_type.string]} : (tensor<*x!tf_type.variant>, tensor<i64>) -> tensor<!tf_type.variant> %1 = "tf.ReduceDataset"(%0, %cst_1, %arg0) { Targuments = [!tf_type.resource], Tstate = [i32], device = "", f = @__reduce_func, f._tf_data_function = true, output_shapes = [#tf_type.shape<>],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 09 11:49:28 UTC 2022 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
llvm::SmallVector<Type, 4> output_types; auto result = tensorflow::GetOutputTypesForLogicalDeviceComputation( core, output_sharding_config, cluster_func, &output_types, &(*cluster_to_core_index)[core]); if (failed(result)) return failure(); for (Type t : output_types) concatenated_output_types.emplace_back(t); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0)