Home
last modified time | relevance | path

Searched refs:ResolveDeviceAssignment (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.h186 StatusOr<absl::optional<xla::DeviceAssignment>> ResolveDeviceAssignment(
Dxla_helpers.cc223 StatusOr<absl::optional<xla::DeviceAssignment>> ResolveDeviceAssignment( in ResolveDeviceAssignment() function
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compile_on_demand_op.cc79 ResolveDeviceAssignment(ctx, result->collective_reduce_info); in Run()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc277 ResolveDeviceAssignment(ctx, compilation_result->collective_reduce_info); in Compute()