Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for set_device_ordinal (0.33 sec)

  1. tensorflow/compiler/jit/device_compiler_client.cc

      if (result.collective_info) {
        build_options.set_num_replicas(result.collective_info->group_size);
      }
      if (options.device_ordinal != -1) {
        build_options.set_device_ordinal(options.device_ordinal);
      } else if (default_device_ordinal != -1) {
        build_options.set_device_ordinal(default_device_ordinal);
      }
      build_options.set_result_layout(result.xla_output_shape);
      build_options.set_device_allocator(options.device_allocator.get());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 26 20:35:26 UTC 2023
    - 1.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/get_compiler_ir.cc

      }
      xla::ExecutableBuildOptions build_options;
      if (result.collective_info) {
        build_options.set_num_replicas(result.collective_info->group_size);
      }
      build_options.set_device_ordinal(
          options.device_ordinal != -1 ? options.device_ordinal
                                       : local_client->default_device_ordinal());
      build_options.set_result_layout(result.xla_output_shape);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 19K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/xla_tpu_device.cc

              }
            });
    
        auto* const dst_device_to_device_stream_impl =
            static_cast<tpu::TpuStreamInterface*>(dst_device_to_device_stream);
    
        const int dst_device_ordinal =
            dst_xla_context->stream()->parent()->device_ordinal();
    
        XlaTensor* const xla_input = XlaTensor::FromTensor(input);
        TF_RET_CHECK(xla_input != nullptr && xla_input->has_shaped_buffer());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 20.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc

                  llvm::dyn_cast<mlir::TF::_XlaRecvAtHostOp>(clone)) {
            recv_at_host.setDeviceOrdinal(core);
            clone->setOperand(0, core_to_compilation_key[core]);
          } else if (auto send_from_host =
                         llvm::dyn_cast<mlir::TF::_XlaSendFromHostOp>(clone)) {
            send_from_host.setDeviceOrdinal(core);
            clone->setOperand(1, core_to_compilation_key[core]);
          } else if (auto recv_at_host =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 68.3K bytes
    - Viewed (0)
Back to top