- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for SplitOp (0.12 sec)
-
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
<< "output #" << i << " should be " << expected_output_type << " instead got " << output.getType(); } return success(); } mlir::LogicalResult SplitOp::verify() { SplitOp op = *this; int64_t num_splits = op.getNumSplits(); if (op.getNumResults() != num_splits) return op.emitOpError("output count should match 'num_splits' attribute");
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/ir/tf_ops_n_z.cc
<< input_rank << ", " << input_rank << ")"; } if (index < 0) index += input_rank; *dim_index = index; return success(); } LogicalResult SplitOp::verify() { SplitOp op = *this; std::optional<int64_t> dim_index; if (failed(VerifySplitInputAndSplitDim(op, &dim_index))) return failure(); if (!dim_index) return success(); int64_t input_dim_size =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_rewrite.mlir
// tile_assignment_devices: 0 // last_tile_dims: REPLICATED // Serialized string: // "\08\03\1A\06\01\01\01\01\01\01\22\01\00B\01\00" // Test that an input sharding with last_tile_dims REPLICATED won't generate SplitOp. //CHECK-NOT: tf.Split
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 172.9K bytes - Viewed (0)