- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for indices_size (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/host_runtime/tfrt_ops.cc
int32_t indices_size = get_resource_op->getAttrOfType<mlir::ArrayAttr>("indices").size(); int32_t shared_name_size = get_resource_op->getAttrOfType<mlir::ArrayAttr>("shared_name").size(); int32_t container_size = get_resource_op->getAttrOfType<mlir::ArrayAttr>("container").size(); if (!(indices_size == shared_name_size && shared_name_size == container_size)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.6K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad.cc
} Output BatchGatherGrad(const Scope& scope, Output params_shape, Output values, Output indices, int batch_dims, Output gather_dim_size) { // Axis is the first non-batch dimension. auto indices_size = ExpandDims(scope, Size(scope, indices), 0); Output outer_shape, flat_values_shape; if (batch_dims != 0) { auto values_shape = Shape(scope, values);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 31.7K bytes - Viewed (0)