OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:original_shape
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/tests/
replay_test.cc
54
std::unique_ptr<ProgramShape>
original_shape
=
local
58
ASSERT_TRUE(protobuf_util::ProtobufEquals(*
original_shape
, *replayed_shape));
86
std::unique_ptr<ProgramShape>
original_shape
=
local
90
ASSERT_TRUE(protobuf_util::ProtobufEquals(*
original_shape
, *replayed_shape));
132
std::unique_ptr<ProgramShape>
original_shape
=
local
136
ASSERT_TRUE(protobuf_util::ProtobufEquals(*
original_shape
, *replayed_shape));
/external/tensorflow/tensorflow/compiler/xla/service/
bfloat16_normalization.cc
194
auto
original_shape
= crs->shape();
local
219
*tuple->mutable_shape() =
original_shape
;
/external/tensorflow/tensorflow/core/kernels/
strided_slice_op.cc
318
TensorShape
original_shape
= old_lhs.shape();
variable
/external/tensorflow/tensorflow/compiler/xla/client/
computation_builder.cc
433
std::unique_ptr<Shape>
original_shape
= shape_or_status.ConsumeValueOrDie();
local
435
VLOG(3) << "original shape: " << ShapeUtil::HumanString(*
original_shape
);
446
for (int i = 0; i < ShapeUtil::Rank(*
original_shape
); ++i) {
448
new_sizes.push_back(
original_shape
->dimensions(i));
450
new_sizes.back() *=
original_shape
->dimensions(i);
[
all
...]
Completed in 1083 milliseconds