OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:ProgramShape
(Results
1 - 25
of
45
) sorted by null
1
2
/external/tensorflow/tensorflow/compiler/aot/
compile.h
36
xla::
ProgramShape
program_shape; // Static shape of args and results.
compile.cc
51
xla::StatusOr<std::unique_ptr<xla::
ProgramShape
>> pshape_or =
58
xla::
ProgramShape
* pshape = &compile_result->program_shape;
codegen.cc
103
const xla::
ProgramShape
& ps = compile_result.program_shape;
153
Status GenArgMethods(const tf2xla::Config& config, const xla::
ProgramShape
& ps,
193
const xla::
ProgramShape
& ps, string* methods) {
299
const xla::
ProgramShape
& ps = compile_result.program_shape;
486
static const xla::
ProgramShape
* StaticProgramShape() {
487
static const xla::
ProgramShape
* kShape = {{PROGRAM_SHAPE_SHIM_EXPRESSION}};
539
std::unique_ptr<xla::
ProgramShape
> program_shape;
543
tensorflow::MakeUnique<xla::
ProgramShape
>(compile_result.program_shape);
557
"xla::
ProgramShape
", program_shape.get()));
/external/tensorflow/tensorflow/compiler/xla/service/
hlo_module_config.cc
33
HloModuleConfig::HloModuleConfig(const
ProgramShape
& program_shape)
37
const
ProgramShape
& program_shape) {
hlo_module_config.h
39
//
ProgramShape
creates a computation layout using this shape.
41
explicit HloModuleConfig(const
ProgramShape
& program_shape);
50
void SetDefaultComputationLayout(const
ProgramShape
& program_shape);
shape_inference.h
86
const
ProgramShape
& to_apply,
136
const
ProgramShape
& to_apply);
142
const
ProgramShape
& to_apply_shape);
147
const Shape& operand_shape, const
ProgramShape
& select_shape,
149
const Shape& init_value_shape, const
ProgramShape
& scatter_shape);
189
static StatusOr<Shape> InferWhileShape(const
ProgramShape
& condition,
190
const
ProgramShape
& body,
196
const Shape& false_operand, const
ProgramShape
& true_computation,
197
const
ProgramShape
& false_computation);
248
const
ProgramShape
& to_apply)
[
all
...]
computation_layout.cc
26
ComputationLayout::ComputationLayout(const
ProgramShape
& program_shape)
computation_layout.h
35
// Constructs a ComputationLayout from a
ProgramShape
. The layouts of the
37
//
ProgramShape
are ignored.
38
explicit ComputationLayout(const
ProgramShape
& program_shape);
compile_only_service.cc
96
std::shared_ptr<const
ProgramShape
> program_shape,
service.h
252
const
ProgramShape
& program_shape,
277
const
ProgramShape
& program_shape,
user_computation.h
277
StatusOr<std::shared_ptr<const
ProgramShape
>> ComputeProgramShape(
402
// Memoized
ProgramShape
and its version. A shared_ptr is used because
405
mutable std::shared_ptr<const
ProgramShape
> program_shape_ GUARDED_BY(mutex_);
/external/tensorflow/tensorflow/compiler/xla/client/
computation.h
65
StatusOr<
ProgramShape
> GetProgramShape() const;
computation.cc
67
StatusOr<
ProgramShape
> Computation::GetProgramShape() const {
client.h
157
StatusOr<std::unique_ptr<
ProgramShape
>> GetComputationShape(
/external/tensorflow/tensorflow/compiler/tf2xla/
xla_compiled_cpu_function.h
28
class
ProgramShape
;
78
const xla::
ProgramShape
* program_shape = nullptr;
208
const xla::
ProgramShape
*
ProgramShape
() const { return program_shape_; }
240
const xla::
ProgramShape
* program_shape_ = nullptr;
xla_jit_compiled_cpu_function.h
82
std::unique_ptr<const xla::
ProgramShape
> program_shape_;
xla_jit_compiled_cpu_function_test.cc
119
const xla::
ProgramShape
* program_shape = function.
ProgramShape
();
xla_jit_compiled_cpu_function.cc
40
const xla::
ProgramShape
& program_shape) {
123
TF_ASSIGN_OR_RETURN(std::unique_ptr<xla::
ProgramShape
> program_shape,
/external/tensorflow/tensorflow/compiler/xla/tests/
replay_test.cc
54
std::unique_ptr<
ProgramShape
> original_shape =
56
std::unique_ptr<
ProgramShape
> replayed_shape =
86
std::unique_ptr<
ProgramShape
> original_shape =
88
std::unique_ptr<
ProgramShape
> replayed_shape =
132
std::unique_ptr<
ProgramShape
> original_shape =
134
std::unique_ptr<
ProgramShape
> replayed_shape =
axpy_simple_test.cc
65
TF_ASSERT_OK_AND_ASSIGN(
ProgramShape
shape, builder.GetProgramShape());
hlo_test_base.cc
63
bool ProgramShapesEqual(const
ProgramShape
& lhs, const
ProgramShape
& rhs) {
75
ProgramShape
GetProgramShapeWithLayout(const HloModule& module) {
76
ProgramShape
program_shape;
/external/tensorflow/tensorflow/compiler/xla/
layout_util.h
59
// Sets the layouts of all Shapes within the given
ProgramShape
to the
61
static void SetToDefaultLayout(
ProgramShape
* program_shape);
75
// Clears the layout on all Shapes within the given
ProgramShape
.
76
static void ClearLayout(
ProgramShape
* program_shape);
129
// Returns whether all Shapes within the given
ProgramShape
have layouts.
130
static bool HasLayout(const
ProgramShape
& program_shape);
/external/tensorflow/tensorflow/compiler/xla/tools/
show_signature.cc
18
// Shows the signature (
ProgramShape
) of binary snapshot proto(s) on the command
56
std::unique_ptr<
ProgramShape
> shape =
dumped_computation_to_operation_list.cc
80
std::unique_ptr<
ProgramShape
> program_shape =
dumped_computation_to_text.cc
55
std::unique_ptr<
ProgramShape
> program_shape =
Completed in 967 milliseconds
1
2