- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for new_output_types (0.37 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_merge_variables_with_execute.cc
// Output types. Skip the original outputs for merged assigns. llvm::SmallVector<Type, 8> new_output_types; int old_output_index = 0; for (const auto& type : execute_launch.getResultTypes()) { if (var_access_info.old_to_new_output_mapping[old_output_index] >= 0) { new_output_types.push_back(type); } ++old_output_index; } // The attributes for merged variable reads and updates.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 17:52:11 UTC 2024 - 27K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
} for (auto size : output_type.getShape()) { new_output_shape.push_back(size); } RankedTensorType new_output_type = RankedTensorType::get(new_output_shape, output_type.getElementType()); auto new_slice = rewriter.create<TFL::SliceOp>( slice_op.getLoc(), new_output_type, reshape_op, new_begin, new_size); // Append a reshape at the bottom.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc
} // Check that total number of outputs from regions match the output types of // the parallel_execute op. const int num_output_types = op.getOperation()->getNumResults(); if (num_output_types != output_index) { return op.emitOpError() << "number of output types (" << num_output_types << ") " << "must match the total number of outputs from all " << "regions (" << output_index << ")."; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.4K bytes - Viewed (0)