Home
last modified time | relevance | path

Searched defs:max_shape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dclip_by_value_op.cc31 const TensorShape max_shape = ctx->InputShape(2); in Compile() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dcomparison_expander.cc115 auto max_shape = max_signed->shape(); in ExpandInstruction() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Ddynamic_shaped_ops.cc195 Shape max_shape = FindMaxShape(absl::MakeSpan(root_shapes_ptrs)); in DynamicConditional() local
/external/tensorflow/tensorflow/core/ops/
Dimage_ops.cc338 ShapeHandle max_shape; in __anonf6f833bb0302() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper_test.cc2417 auto max_shape = ops::Placeholder::Shape({64, 64}); in RunTest() local