Home
last modified time | relevance | path

Searched defs:minval_shape (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Drandom_ops.cc91 const TensorShape minval_shape = ctx->InputShape(1); in Compile() local
H A Dstateless_random_ops.cc205 TensorShape minval_shape = ctx->InputShape(2); in Compile() local
H A Dstateless_random_ops_v2.cc307 TensorShape minval_shape = ctx->InputShape(minval_input_idx); in Compile() local