Home
last modified time | relevance | path

Searched defs:shape_arg (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dservice.cc921 const Shape& shape_arg = replicated_arguments.front()[i]->on_device_shape(); in Execute() local
/external/pytorch/torch/csrc/jit/passes/
Dsymbolic_shape_analysis.cpp446 auto shape_arg = shape_arguments->at(*norm_index); in substituteSymbolicProperties() local