Searched refs:valid_bitcast_callback_ (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dalgebraic_simplifier.h46 valid_bitcast_callback_(std::move(valid_bitcast_callback)),
58 ValidBitcastCallback valid_bitcast_callback_; member in class:xla::AlgebraicSimplifier
H A Dalgebraic_simplifier.cc192 valid_bitcast_callback_(std::move(valid_bitcast_callback)),
298 AlgebraicSimplifier::ValidBitcastCallback valid_bitcast_callback_; member in class:xla::AlgebraicSimplifierVisitor
1544 ReshapeIsBitcast(reshape, valid_bitcast_callback_)) {
1972 if (!valid_bitcast_callback_(input_shape, new_input_shape) ||
1973 !valid_bitcast_callback_(filter_shape, new_filter_shape) ||
1974 !valid_bitcast_callback_(dot_output_shape, convolution_shape)) {
2071 comp, is_layout_sensitive_, valid_bitcast_callback_,

Completed in 162 milliseconds