Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 12 of 12 for device_var_reads_indices (1.81 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

                   << entry.index();
        }
    
        return success();
      };
    
      return failure(
          failed(check_attr(op.getDeviceVarReadsIndices(),
                            /*name=*/"device_var_reads_indices", /*min=*/0)) ||
          failed(check_attr(op.getDeviceVarUpdatesIndices(),
                            /*name=*/"device_var_updates_indices", /*min=*/-1)));
    }
    
    void TPUExecuteAndUpdateVariablesOp::getEffects(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

      let description = [{
    It (optionally) reads device variables, loads and executes a TPU program on a
    TPU device, and then (optionally) in-place updates variables using the program
    outputs, as specified in attributes device_var_reads_indices (program input
    indices from directly reading variables) and device_var_updates_indices (program
    output indices used to update variables, -1 means no-update/read-only). Such
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top