Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for GatherElements (0.2 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.h

    // Adds two values, or creates a logical-or if they are boolean type.
    Value AccumulateBuffers(Value a, Value b, OpBuilder builder, Location loc);
    
    // Gathers elements in buffer with the indices.
    Value GatherElements(Value indices, Value buffer, OpBuilder builder,
                         Location loc);
    
    // Scatters elements into buffer, where each scattered element is accumulated
    // with the old value in buffer.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 02 20:41:19 UTC 2023
    - 4.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc

          first_index = last_index;
          continue;
        }
        if (last_index + 1 != ind.getSExtValue()) return -1;
        last_index++;
      }
      return first_index;
    }
    
    }  // namespace
    
    Value GatherElements(Value indices, Value buffer, OpBuilder builder,
                         Location loc) {
      auto buffer_type = mlir::cast<RankedTensorType>(buffer.getType());
      auto result_shape = llvm::to_vector<8>(buffer_type.getShape());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

      auto it = buffer_to_size.find(gather.getInputHandle());
      if (it == buffer_to_size.end()) {
        return gather.emitOpError("unknown tensor list");
      }
      auto buffer = gather.getInputHandle();
      auto result = cutil::GatherElements(gather.getIndices(), buffer,
                                          OpBuilder(gather), gather.getLoc());
      gather.getValues().replaceAllUsesWith(result);
      gather.erase();
      return success();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc

        return gather.emitOpError("unknown tensor array");
      }
      OpBuilder builder(gather);
      auto buffer = cutil::ReadLocalVariable(local_var, builder, gather.getLoc());
      auto result = cutil::GatherElements(gather.getIndices(), buffer, builder,
                                          gather.getLoc());
      ReplaceAllUsesExceptTerminator(gather.getValue(), result);
      ReplaceAllUsesWithCast(gather.getValue(), result);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 02 20:41:19 UTC 2023
    - 40.2K bytes
    - Viewed (0)
Back to top