Home
last modified time | relevance | path

Searched defs:offset_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_batch_normalization.cc97 const auto& offset_shape = offset_array.shape(); in Run() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander_test.cc59 Shape offset_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
Dshape_inference.cc1201 const Shape& offset_shape, int64_t feature_index) { in InferBatchNormTrainingShape()
1306 const Shape& offset_shape, const Shape& mean_shape, in InferBatchNormInferenceShape()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2373 const Shape& offset_shape = per_partition_size_hlo->shape(); in HandleDynamicUpdateSlice() local