- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 215 for tpu0 (0.2 sec)
-
tensorflow/compiler/mlir/tfrt/tests/fuse_tpu_compile_and_execute_ops.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 13.8K bytes - Viewed (0) -
hack/make-rules/make-help.sh
# See the License for the specific language governing permissions and # limitations under the License. set -o errexit set -o nounset set -o pipefail red=$(tput setaf 1) reset=$(tput sgr0) readonly red reset KUBE_ROOT=$(dirname "${BASH_SOURCE[0]}")/../.. ALL_TARGETS=$(make -C "${KUBE_ROOT}" PRINT_HELP=y -rpn | sed -n -e '/^$/ { n ; /^[^ .#][^ ]*:/ { s/:.*$// ; p ; } ; }' | sort)
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Thu Oct 20 06:46:18 UTC 2022 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_validate_inputs.cc
if (!op->hasAttr(TF::kTpuReplicateAttr)) { op->emitOpError("TF2XLA TPU bridge input check: " + errormsg() + "missing _tpu_replicate attr"); return false; } auto opattr = op->getAttr(TF::kTpuReplicateAttr); if (opattr != attr) { op->emitOpError("TF2XLA TPU bridge input check: " + errormsg() + "invalid _tpu_replicate attr.")
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/quantization/tensorflow/python/pywrap_quantize_model.cc
if (!exported_model.ok()) return exported_model.status(); // Remove the `tpu` tag from the debug quantized saved model as it is // for CPU. Note the 'tpu' value should be the same as `TPU` defined in // tensorflow/python/saved_model/tag_constants.py. if (quantization_options.has_debugger_config()) { tags.erase("tpu"); } py_function_library.SaveExportedModel(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 09 06:33:29 UTC 2024 - 12K bytes - Viewed (0) -
ci/official/containers/linux_arm64/devel.usertools/aarch64_clang.bazelrc
# bazel test invocation as normal. test:nonpip_filters --test_tag_filters=-no_oss,-oss_serial,-gpu,-tpu,-benchmark-test,-v1only,-no_aarch64,-no_oss_py38,-no_oss_py39,-no_oss_py310 test:nonpip_filters --build_tag_filters=-no_oss,-oss_serial,-gpu,-tpu,-benchmark-test,-v1only,-no_aarch64,-no_oss_py38,-no_oss_py39,-no_oss_py310
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 21 12:25:39 UTC 2023 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.h
mlir::ArrayAttr device_assignment_attr); // Finds the TPU compilation device and execution devices from `devices` for a // TPU computation subgraph. Compilation device is determined from looking up // all TPU_SYSTEM:0 devices and choosing the CPU device associated to the first // TPU_SYSTEM device sorted lexicographically by replica and task. Execution // devices are determined by looking up all TPU devices associated with each
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 11.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_cluster_util.cc
// symbol_table caches callees in the CallGraph. SymbolTableCollection symbol_table; // List pending nodes to traverse with their root TPU cluster. llvm::SmallVector<std::pair<CallGraphNode*, tf_device::ClusterOp>> pending_call_nodes; // Cache the host device for each TPU cluster. std::unordered_map<Operation*, std::optional<std::string>> cluster_to_host; auto insert_pending_op = [&](Operation* op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 04:50:13 UTC 2023 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_update_embedding_enqueue_op_inputs.cc
}); return failure(walk_result.wasInterrupted()); } // Updates the operand of TPU embedding enqueue ops depending on whether // the graph is in training mode or in non-training mode. // If SendTPUEmbeddingGradients op is present, this means that graph is in // training mode. As so, correctly feed in `then` branch value of SelectV2 // operand as inputs to the TPU embedding enqueue ops. LogicalResult UpdateEmbeddingEnqueueOpInput(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 7.7K bytes - Viewed (0) -
src/internal/types/testdata/spec/assignability.go
} // "x's type V is not a named type and T is a type parameter, and x is assignable to each specific type in T's type set." func _[ TP0 any, TP1 ~_Chan, TP2 ~chan int | ~chan byte, ]() { var ( _ TP0 = c // ERRORx `cannot use c .* as TP0 value` _ TP0 = C // ERRORx `cannot use C .* as TP0 value` _ TP1 = c _ TP1 = C // ERRORx `cannot use C .* as TP1 value`
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Thu Nov 09 17:24:42 UTC 2023 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/legalize_tf_quant_test.cc
#include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/protobuf/config.pb.h" #include "tensorflow/core/protobuf/tpu/compile_metadata.pb.h" #include "tensorflow/core/tpu/kernels/tpu_compile_op_support.h" #include "tsl/lib/core/status_test_util.h" namespace mlir::quant::stablehlo { namespace { using ::testing::Test;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 18:43:55 UTC 2024 - 7.2K bytes - Viewed (0)