- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 87 for Indices (0.12 sec)
-
tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/constants.mlir
%0 = "tfl.pseudo_sparse_const"() {compressed_data = dense<[1.0, 2.0, 0.5, 0.25,...
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 12.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/constants_offset.mlir
%0 = "tfl.pseudo_sparse_const"() {compressed_data = dense<[1.0, 2.0, 0.5, 0.25,...
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 12.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/scatter.h
// // Eg 1. An update in canonical form: // * indices shape(A,B,C) // * updates shape(A,B,D,E,F) // Then: // * D,E,F are the update window dims [2,3,4] // * C is the index vector dimension // * A,B iterate over the updates and indices // // If `update_window_dims` are not the trailing dimensions then updates must be // transposed. // // Eg 2. An update in non-canonical form: // * indices shape(a,b,c) // * updates shape(d,e,f,a,b)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.td
def LowerScatterNdOp : Pat<(TF_ScatterNdOp $indices, TensorOf<[AnyInteger, AnyFloat, AnyComplex]>:$updates, $shape, ConstantStrAttr<StrAttr, "">), (TF_TensorScatterAddOp (TF_FillOp $shape, (TF_ConstOp (GetScalarOfType<0> $updates))), $indices, $updates)>; def LowerScatterNdOpDefaultBadIndicesPolicy : Pat<(TF_ScatterNdOp $indices,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 04 13:30:42 UTC 2024 - 24.7K bytes - Viewed (0) -
tensorflow/compiler/aot/codegen.cc
indices += absl::StrCat("[dim", dim, "]"); count *= shape.dimensions(dim); } } rewrites->push_back({"{{I}}", absl::StrCat(i)}); rewrites->push_back({"{{TYPE}}", type}); rewrites->push_back({"{{DIM_VARS}}", absl::StrJoin(dim_vars, ", ")}); rewrites->push_back({"{{DIM_SIZES}}", dim_sizes}); rewrites->push_back({"{{INDICES}}", indices});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 01:20:01 UTC 2024 - 36.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc
ArrayRef<Value>{a, b}); } namespace { int64_t GetFirstIfIndicesAreContiguous(Value indices) { auto type = mlir::dyn_cast<RankedTensorType>(indices.getType()); if (!type) return -1; auto indices_op = indices.getDefiningOp(); if (!indices_op) return -1; auto const_op = llvm::dyn_cast<TF::ConstOp>(indices_op); if (!const_op) return -1;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.5K bytes - Viewed (0) -
tensorflow/compiler/aot/codegen_test_h.golden
// // T* argN_data() // Returns the buffer of type T for positional argument N. // // T& argN(...dim indices...) // Returns a reference to the value of type T for positional argument N, // with dim indices specifying which value. No bounds checking is performed // on dim indices. void set_arg0_data(const void* data) { set_arg_data(0, data); } float* arg0_data() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 01:20:01 UTC 2024 - 16.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/util.cc
indices = rewriter.create<mhlo::ReshapeOp>(parent_op->getLoc(), indices_type, indices); } else if (index_vector_dim != indices_type.getRank() - 1) { // If index_vector_dim isn't the last dimension in indices then it isn't // supported yet. // TODO(tberghammer): Transpose indices to support this usecase.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_patterns.td
def LegalizeGather: Pat< (TF_GatherOp $params, $indices, $ignored_validate_indices), (TFL_GatherOp $params, $indices, ConstantAttr<I32Attr, "0">, ConstantAttr<I32Attr, "0">)>; def LegalizeGatherNd : Pat<(TF_GatherNdOp $params, $indices, $bad_indices_policy), (TFL_GatherNdOp $params, $indices)>; def LegalizeGatherV2 : Pat< (TF_GatherV2Op $params, $indices, (Arith_ConstantOp ElementsAttr:$axis), $batch_dims),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 04 13:30:42 UTC 2024 - 28.5K bytes - Viewed (0) -
src/go/types/check_test.go
continue } } indices = append(indices, i) } if len(indices) == 0 { t.Errorf("%s: no error expected: %q", gotPos, gotMsg) continue } // len(indices) > 0 // If there are multiple matching errors, select the one with the closest column position. index := -1 // index of matching error var delta int for _, i := range indices {
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Wed May 22 19:45:33 UTC 2024 - 14.1K bytes - Viewed (0)