OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:arg_shape
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/
shape_inference.cc
398
const Shape*
arg_shape
= nullptr;
local
403
if (!
arg_shape
) {
404
arg_shape
= shape;
405
element_type =
arg_shape
->element_type();
408
if (ShapeUtil::Rank(*
arg_shape
) != ShapeUtil::Rank(*shape)) {
412
ShapeUtil::Rank(*
arg_shape
),
413
ShapeUtil::HumanString(*
arg_shape
).c_str(), ShapeUtil::Rank(*shape),
416
if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(*
arg_shape
, *shape)) {
419
PrimitiveType_Name(
arg_shape
->element_type()).c_str(),
423
dimension_number < ShapeUtil::Rank(*
arg_shape
); ++dimension_number)
1089
const Shape*
arg_shape
=
arg_shape
s[0];
local
2437
const Shape&
arg_shape
= *
arg_shape
s[i];
local
[
all
...]
/external/tensorflow/tensorflow/compiler/tf2xla/
xla_jit_compiled_cpu_function.cc
45
const xla::Shape&
arg_shape
= program_shape.parameters(i);
local
47
arg_sizes.push_back(xla::ShapeUtil::ByteSizeOf(
arg_shape
, kPointerSize));
Completed in 79 milliseconds