OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:in0_shape
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
cross_op.cc
28
TensorShape
in0_shape
= ctx->InputShape(0);
variable
30
OP_REQUIRES(ctx,
in0_shape
== in1_shape,
32
in0_shape
.DebugString(), " vs. ",
34
OP_REQUIRES(ctx,
in0_shape
.dims() >= 1,
36
in0_shape
.DebugString()));
38
auto inner_dim =
in0_shape
.dim_size(
in0_shape
.dims() - 1);
49
std::vector<int64> starts(
in0_shape
.dims(), 0);
51
for (auto dim_size :
in0_shape
.dim_sizes()) {
54
std::vector<int64> strides(
in0_shape
.dims(), 1)
[
all
...]
mirror_pad_op.cc
81
xla::StatusOr<std::unique_ptr<xla::Shape>>
in0_shape
= b->GetShape(in0);
variable
82
OP_REQUIRES(ctx,
in0_shape
.ok(),
in0_shape
.status());
84
DoMirrorPad(in0, *
in0_shape
.ValueOrDie(), pad_literal, b);
Completed in 77 milliseconds